diff --git a/main.py b/main.py
index 6359d36..5c25d96 100644
--- a/main.py
+++ b/main.py
@@ -52,6 +52,7 @@ with gr.Blocks(theme=set_theme, analytics_enabled=False) as demo:
                     with gr.Row():
                         resetBtn = gr.Button("重置", variant="secondary")
                         submitBtn = gr.Button("提交", variant="primary")
+                        stopBtn = gr.Button("停止", variant="stop")
             with gr.Row():
                 from check_proxy import check_proxy
                 statusDisplay = gr.Markdown(f"Tip: 按Enter提交, 按Shift+Enter换行. \nNetwork: {check_proxy(proxies)}\nModel: {LLM_MODEL}")
@@ -79,9 +80,10 @@ with gr.Blocks(theme=set_theme, analytics_enabled=False) as demo:
 
     txt.submit(**predict_args)
     txt.submit(**empty_txt_args)
-    submitBtn.click(**predict_args)
+    submit_event = 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:
         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)