mirror of https://github.com/xtekky/gpt4free
Create image_models.py
parent
2b9a95c5b3
commit
3f17f9b8e2
@ -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
|
Loading…
Reference in New Issue