Update app.py

app.py

@@ -47,6 +47,8 @@ def respond(
     # Call Hugging Face model for response
     try:
         responses = []
+        generated_image = None
+
         for response in client.chat_completion(
             messages,
             max_tokens=max_tokens,
@@ -61,10 +63,9 @@ def respond(
             if response.choices[0].delta.image:
                 image_b64 = response.choices[0].delta.image
                 image_data = base64.b64decode(image_b64)
-                image = Image.open(io.BytesIO(image_data))
-                yield responses, image
-            else:
-                yield responses, None
+                generated_image = Image.open(io.BytesIO(image_data))
+
+        yield responses, generated_image
 
     except Exception as e:
         yield [str(e)], None