Update app.py

app.py

@@ -210,4 +210,24 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     generate_btn.click(
         process_and_save_image,
-        inputs=[height, width, steps, scales, prompt, seed],
+        inputs=[height, width, steps, scales, prompt, seed],
+
+
+    generate_btn.click(
+        process_and_save_image,
+        inputs=[height, width, steps, scales, prompt, seed],
+        outputs=output
+    )
+    
+    randomize_seed.click(
+        update_seed,
+        outputs=[seed]
+    )
+    
+    generate_btn.click(
+        update_seed,
+        outputs=[seed]
+    )
+
+if __name__ == "__main__":
+    demo.launch(allowed_paths=[PERSISTENT_DIR])