Update app.py

app.py

@@ -120,6 +120,7 @@ with gr.Blocks() as app:
                 height="auto")
             form_start_btn = gr.Button("Start Harmonization", interactive=False)
             form_reset_btn = gr.Button("Reset", interactive=True)
+            form_stop_btn = gr.Button("Stop", interactive=True)
 
 
     def on_change_form_composite_image(form_composite_image):
@@ -254,6 +255,12 @@ with gr.Blocks() as app:
     form_reset_btn.click(on_click_form_reset_btn,
                          inputs=None, outputs=[form_log, form_composite_image, form_mask_image, form_start_btn])
 
+    def on_click_form_stop():
+        gr.close_all()
+
+    form_stop_btn.click(on_click_form_stop,
+                         inputs=None, outputs=None)
+
     gr.Markdown("""
         ## Quick Start
         1. Select desired `Pretrained Model`.