From 2736189eb4a1394909bf82aa67d0f33cb1b28f2d Mon Sep 17 00:00:00 2001 From: Baize Date: Mon, 3 Apr 2023 04:21:00 +0000 Subject: [PATCH] Update app.py --- app.py | 27 ++++++++++++++++++--------- 1 file changed, 18 insertions(+), 9 deletions(-) diff --git a/app.py b/app.py index ef9200d..fb28618 100644 --- a/app.py +++ b/app.py @@ -107,6 +107,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): @@ -191,9 +193,9 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo: fn=transfer_input, inputs=[user_input], outputs=[user_question, user_input, submitBtn], show_progress=True ) - user_input.submit(**transfer_input_args).then(**predict_args) + predict_event1 = user_input.submit(**transfer_input_args).then(**predict_args) - submitBtn.click(**transfer_input_args).then(**predict_args) + predict_event2 = submitBtn.click(**transfer_input_args).then(**predict_args) emptyBtn.click( reset_state, @@ -202,7 +204,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo: ) emptyBtn.click(**reset_args) - retryBtn.click(**retry_args) + predict_event3 = retryBtn.click(**retry_args) delLastBtn.click( delete_last_conversation, @@ -210,11 +212,18 @@ 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 + ] + ) demo.title = "Baize" -if __name__ == "__main__": - reload_javascript() - demo.queue(concurrency_count=1).launch( - share=False, favicon_path="./assets/favicon.ico", inbrowser=True - ) +demo.queue( + concurrency_count=1, + max_size=100, +).launch( + max_threads=5, + server_name="0.0.0.0", +) \ No newline at end of file