Update app.py

app.py

@@ -64,6 +64,8 @@ def respond(
                 image_b64 = response.choices[0].delta.image
                 image_data = base64.b64decode(image_b64)
                 generated_image = Image.open(io.BytesIO(image_data))
+                # Optionally convert to RGB if needed
+                # generated_image = generated_image.convert("RGB")
 
         yield responses, generated_image
 
@@ -101,4 +103,3 @@ try:
 
 except Exception as e:
     print(f"Error during Gradio setup: {str(e)}")
-