diff --git a/app.py b/app.py
index 742d965..9d5e770 100644
--- a/app.py
+++ b/app.py
@@ -119,7 +119,8 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     ).style(container=False)
                 with gr.Column(min_width=70, scale=1):
                     submitBtn = gr.Button("Send")
-                  
+                with gr.Column(min_width=70, scale=1):
+                    cancelBtn = gr.Button("Stop")
             with gr.Row(scale=1):
                 emptyBtn = gr.Button(
                     "🧹 New Conversation",
@@ -156,7 +157,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     )
                     max_context_length_tokens = gr.Slider(
                         minimum=0,
-                        maximum=2048,
+                        maximum=4096,
                         value=2048,
                         step=128,
                         interactive=True,
@@ -221,12 +222,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         [chatbot, history, status_display],
         show_progress=True,
     )
-    #cancelBtn.click(
-    #    cancel_outputing, [], [status_display], 
-    #    cancels=[
-    #        predict_event1,predict_event2,predict_event3
-    #    ]
-    #)    
+    cancelBtn.click(
+        cancel_outputing, [], [status_display], 
+        cancels=[
+            predict_event1,predict_event2,predict_event3
+        ]
+    )    
 demo.title = "Baize"
 
 demo.queue(concurrency_count=1).launch()
\ No newline at end of file