diff --git a/main.py b/main.py
index 8a6db8f..5f02a38 100644
--- a/main.py
+++ b/main.py
@@ -84,6 +84,7 @@ with gr.Blocks(theme=set_theme, analytics_enabled=False) as demo:
     can_cancel_events.append(submitBtn.click(**predict_args))
     submitBtn.click(**empty_txt_args)
     resetBtn.click(lambda: ([], [], "已重置"), None, [chatbot, history, statusDisplay])
+    stopBtn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
     for k in functional:
         click_handle = functional[k]["Button"].click(predict,
             [txt, top_p, temperature, chatbot, history, system_prompt, gr.State(True), gr.State(k)], [chatbot, history, statusDisplay], show_progress=True)