Update app.py

app.py

@@ -195,6 +195,7 @@ with gr.Blocks() as app:
     # The rest of your code for setting up the app
     gallery.select(update_selection, outputs=[prompt, selected_info, selected_state])
     prompt.submit(fn=run_lora, inputs=[prompt, selected_state], outputs=[result])
+    button.click(fn=run_lora, inputs=[prompt, selected_state], outputs=[result])
     post_process_button.click(
     fn=postprocess, 
     inputs=[