Update app.py

app.py

@@ -236,13 +236,16 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
         inputs=[input_box]
     )
     # Generate button logicg
-    def generate_and_display_images():
-        image_paths = generate_top_dresses(prompt)
-        return image_paths + image_paths  # One for display, one for hidden path tracking
+    def generate_and_display_images(prompt):
+        images, paths = generate_top_dresses(prompt)
+        return images + paths
+
         
     submit_button.click(
         fn=generate_and_display_images,
+        inputs=[input_box],
         outputs=image_components + file_paths
     )
 
+
 demo.launch()