From 9aeae65b9b0e7322f63186c0df061de06ca642ee Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Sun, 11 Feb 2024 09:26:02 +0100
Subject: Improve preview in image generation of Gemini Fix issue in Bing Image
 Decorator

---
 g4f/Provider/create_images.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'g4f/Provider/create_images.py')

diff --git a/g4f/Provider/create_images.py b/g4f/Provider/create_images.py
index b8bcbde3..9a9e3f08 100644
--- a/g4f/Provider/create_images.py
+++ b/g4f/Provider/create_images.py
@@ -87,7 +87,7 @@ class CreateImagesProvider(BaseProvider):
         messages.insert(0, {"role": "system", "content": self.system_message})
         buffer = ""
         for chunk in self.provider.create_completion(model, messages, stream, **kwargs):
-            if buffer or "<" in chunk:
+            if isinstance(chunk, str) and buffer or "<" in chunk:
                 buffer += chunk
                 if ">" in buffer:
                     match = re.search(r'<img data-prompt="(.*?)">', buffer)
-- 
cgit v1.2.3