Update app.py

app.py

@@ -75,8 +75,9 @@ def chat_fn(message, chat_history, seed, randomize_seed, guidance_scale, steps,
     # Move model back to CPU to free GPU memory
     pipe = pipe.to("cpu")
     torch.cuda.empty_cache()
-        
-    return image
+    
+    # Return the image in the correct format for ChatInterface
+    return {"files": [image], "text": None}
 
 # --- UI Definition using gr.ChatInterface ---