Update app.py

app.py

@@ -91,11 +91,29 @@ if torch.cuda.is_available():
 
         print(image_paths)
         return image_paths, seed, download_links
+
+    if run_button:
+        image_paths, seed, download_links = generate(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            use_negative_prompt=use_negative_prompt,
+            seed=seed,
+            width=width,
+            height=height,
+            guidance_scale=guidance_scale,
+            randomize_seed=randomize_seed,
+        )
+        
+        for image_path in image_paths:
+            st.image(image_path, caption=prompt)
+        
+        for download_link in download_links:
+            st.markdown(download_link, unsafe_allow_html=True)
+        
+        st.text(f"Seed: {seed}")
 else:
     st.warning("CUDA is not available. The demo may not work on CPU.")
 
-    
-
 examples = [
 "a modern hospital room with advanced medical equipment and a patient resting comfortably",
 "a team of surgeons performing a delicate operation using state-of-the-art surgical robots",