Updates

app.py

@@ -230,8 +230,8 @@ with gr.Blocks() as demo:
 
     # Event handlers
     gallery.select(update_selection, [width, height], [prompt, selected_lora, gr.State(), width, height])
-    randomize_seed.change(lambda x: gr.update(visible=not x), randomize_seed, seed_input)
-    generate.click(run_lora, inputs=[prompt, image_input, image_strength, cfg_scale, steps, gr.State(), randomize_seed, seed_input, width, height, lora_scale], outputs=[result, seed_output, progress_bar])
-    cancel.click(lambda: None, None, None, cancels=[generate])
+randomize_seed.change(lambda x: gr.update(visible=not x), randomize_seed, seed_input)
+generate_event = generate.click(run_lora, inputs=[prompt, image_input, image_strength, cfg_scale, steps, gr.State(), randomize_seed, seed_input, width, height, lora_scale], outputs=[result, seed_output, progress_bar])
+cancel.click(lambda: None, None, None, cancels=[generate_event])
 
 demo.queue().launch()