Update app.py

app.py

@@ -292,6 +292,7 @@ with gr.Blocks(css=css) as demo:
                                                  value="3d, cartoon, anime, (deformed eyes, nose, ears, nose), bad anatomy, ugly")
                 with gr.Column():
                     text_button = gr.Button("Generate", variant='primary', elem_id="generate")
+                    stop_btn = gr.Button("Cancel")
 
             with gr.Row():
                 with gr.Column(scale=3):
@@ -382,9 +383,10 @@ with gr.Blocks(css=css) as demo:
         with gr.Tab("Gallery"):
             gallery_obj = gr.Gallery(height=1000, columns=5)
 
-        text_button.click(flip_text,
+        generation_event = text_button.click(flip_text,
                           inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, batch_size, batch_count,
                                   gallery_obj], outputs=[image_output, gallery_obj])
+        stop_btn.click(fn=None, outputs=None, cancels=[generation_event])
         image_input.upload(get_exif_data, inputs=[image_input], outputs=exif_output)
         send_to_txt2img_btn.click(send_to_txt2img, inputs=[image_input],
                                   outputs=[tabs, prompt, negative_prompt, steps, seed,