Update app.py

app.py

@@ -221,4 +221,11 @@ with gr.Blocks(css=css) as demo:
     uploaded_image.change(fn=image2image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
     image_url.submit(fn=image2image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
 
+    gr.on(
+        triggers=[run2_button.click, prompt2.submit],
+        fn=infer,
+        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        outputs=[result, seed]
+    )
+
 demo.launch()