Update app.py

app.py

@@ -49,6 +49,8 @@ with gr.Blocks() as demo:
 
         txt_input = gr.Textbox(label = 'Prompt text')
         gen_button = gr.Button('Generate')
+        stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
+        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
            
         with gr.Row():
             output = [gr.Image(label = m) for m in default_models]
@@ -58,7 +60,8 @@ with gr.Blocks() as demo:
             model_choice.change(extend_choices, model_choice, current_models)
 
             for m, o in zip(current_models, output):
-                gen_button.click(gen_fn, [m, txt_input], o)
+                gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+                stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
                     
             
     with gr.Tab('Single model'):