diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-05-19 17:49:06 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-19 17:49:06 +0200 |
| commit | fea056dd49a2939cb4f0d87ba0b663cc3bf19d57 (patch) | |
| tree | d4f7751e0e7a0093f6110ddae29befadd8c98dfe /g4f | |
| parent | Update async_client.py (diff) | |
| download | gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar.gz gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar.bz2 gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar.lz gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar.xz gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.tar.zst gpt4free-fea056dd49a2939cb4f0d87ba0b663cc3bf19d57.zip | |
Diffstat (limited to 'g4f')
| -rw-r--r-- | g4f/client/client.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/g4f/client/client.py b/g4f/client/client.py index 1b090981..acf53c70 100644 --- a/g4f/client/client.py +++ b/g4f/client/client.py @@ -125,9 +125,12 @@ def iter_image_response(response: Iterator) -> Union[ImagesResponse, None]: return ImagesResponse([Image(image) for image in chunk.get_list()]) def create_image(client: Client, provider: ProviderType, prompt: str, model: str = "", **kwargs) -> Iterator: - prompt = f"create a image with: {prompt}" - if provider.__name__ == "You": + + + if isinstance(provider, type) and provider.__name__ == "You": kwargs["chat_mode"] = "create" + else: + prompt = f"create a image with: {prompt}" return provider.create_completion( model, [{"role": "user", "content": prompt}], |