From 16e8683ec194781c752f7051d9b11125cde7b7cf Mon Sep 17 00:00:00 2001 From: Baize Date: Thu, 6 Apr 2023 07:01:31 +0000 Subject: [PATCH] Update app.py --- app.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) 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