1 files changed, 11 insertions, 0 deletions

diff --git a/g4f/client/image_models.py b/g4f/client/image_models.py
new file mode 100644
index 00000000..0c8ea894
--- /dev/null
+++ b/g4f/client/image_models.py
@@ -0,0 +1,11 @@
+class ImageModels():
+    gemini = Gemini
+    openai = OpenaiChat
+    you = You
+
+    def __init__(self, client: Client) -> None:
+        self.client = client
+        self.default = BingCreateImages(proxy=self.client.get_proxy())
+
+    def get(self, name: str, default: ImageProvider = None) -> ImageProvider:
+        return getattr(self, name) if hasattr(self, name) else default or self.default