Update app.py

app.py

@@ -123,6 +123,8 @@ from PIL import Image
 import numpy as np
 import pickle
 import gradio as gr
+import tempfile
+
 
 # Force CPU usage for optimization
 device = torch.device("cpu")
@@ -167,7 +169,14 @@ def rank_by_clip(images, prompt, top_k=3):  # Reduce top_k for speed
 def generate_top_dresses(prompt):
     _, images = generate_images(G, num_images=20)
     top_images = rank_by_clip(images, prompt, top_k=2)
-    return top_images
+
+    file_paths = []
+    for i, img in enumerate(top_images):
+        temp_path = tempfile.NamedTemporaryFile(suffix=".png", delete=False).name
+        img.save(temp_path)
+        file_paths.append(temp_path)
+
+    return top_images, file_paths
 
 # Launch Gradio
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo: