Update app.py

app.py

@@ -111,6 +111,13 @@ with gr.Blocks(fill_width=True, css=css) as demo:
         outputs=[lora_A],
         queue=True,
     )
+    gr.on(
+        triggers=[generate_button.click],
+        fn=generate_image,
+        inputs=[prompt, num_steps, guidance_scale, sampler, img_width, img_height],
+        outputs=output_image,
+        queue=True,
+    )
     gr.on(
         triggers=[search_civitai_submit.click, search_civitai_query.submit, search_civitai_user.submit],
         fn=search_civitai,