From 0937f373885800513e37b75442067e2f6026b0a8 Mon Sep 17 00:00:00 2001 From: binary-husky Date: Thu, 14 Sep 2023 11:02:40 +0800 Subject: [PATCH] =?UTF-8?q?Predict=E6=8C=89=E9=92=AE=E5=8F=82=E6=95=B0?= =?UTF-8?q?=E4=BF=AE=E6=AD=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index ff4bc78..bfa04f0 100644 --- a/main.py +++ b/main.py @@ -159,7 +159,7 @@ def main(): # 整理反复出现的控件句柄组合 input_combo = [cookies, max_length_sl, md_dropdown, txt, txt2, top_p, temperature, chatbot, history, system_prompt, plugin_advanced_arg] output_combo = [cookies, chatbot, history, status] - predict_args = dict(fn=ArgsGeneralWrapper(predict), inputs=input_combo, outputs=output_combo) + predict_args = dict(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True)], outputs=output_combo) # 提交按钮、重置按钮 cancel_handles.append(txt.submit(**predict_args)) cancel_handles.append(txt2.submit(**predict_args))