Update app.py

app.py

@@ -115,16 +115,19 @@ with gr.Blocks(css="style.css", fill_height=True) as demo:
         value=1.2,
     )
 
-    complete_button.click(
+    # Set up the generation event
+    generation_event = complete_button.click(
         generate,
         inputs=[text_box, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
         outputs=[text_box],
     )
+
+    # Set up the stop event
     stop_button.click(
         None,
         None,
         None,
-        cancels=[complete_button.click]
+        cancels=[generation_event]
     )
 
 if __name__ == "__main__":