From fea056dd49a2939cb4f0d87ba0b663cc3bf19d57 Mon Sep 17 00:00:00 2001
From: H Lohaus <hlohaus@users.noreply.github.com>
Date: Sun, 19 May 2024 17:49:06 +0200
Subject: Update client.py

---
 g4f/client/client.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'g4f/client')

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}],
-- 
cgit v1.2.3