Update app.py

app.py

@@ -54,7 +54,7 @@ with gr.Blocks() as demo:
         txt_input = gr.Textbox(label='Your prompt:', lines=4)
         gen_button = gr.Button('Generate up to 2 images in up to 20 seconds')
         #stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
+        gen_button.click(lambda s: gr.update(interactive = True), None)
         gr.HTML(
         """
             <div style="text-align: center; max-width: 1200px; margin: 0 auto;">