Update app.py

app.py

@@ -77,12 +77,8 @@ def chat_fn(message, chat_history, seed, randomize_seed, guidance_scale, steps,
     pipe = pipe.to("cpu")
     torch.cuda.empty_cache()
     
-    # Save image to temporary file and return the path
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
-        image.save(tmp_file.name, format="PNG")
-        temp_path = tmp_file.name
-    
-    return temp_path
+    # Return the PIL Image as a Gradio Image component
+    return gr.Image(value=image)
 
 # --- UI Definition using gr.ChatInterface ---