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