Update app.py

app.py

@@ -36,35 +36,25 @@ def respond(message, language, task, system_message, max_tokens, temperature, to
     )
     # 解码回复
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    yield response
+    return response
 
 # 更新任务选项的函数
 def update_tasks(language):
     return gr.update(choices=list(options[language].keys()))
 
 # 创建Gradio接口
-demo = gr.ChatInterface(
-    respond,
-    inputs=[
-        gr.Textbox(label="Input Text"),
-        gr.Dropdown(label="Language", choices=list(options.keys()), value="English"),
-        gr.Dropdown(label="Task", choices=list(options['English'].keys())),
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)"
-        ),
-    ],
-    live=True
-)
-
-# 设置语言选择框的动态更新
-demo.components[1].change(update_tasks, inputs=demo.components[1], outputs=demo.components[2])
+with gr.Blocks() as demo:
+    system_message = gr.Textbox(value="You are a friendly Chatbot.", label="System message")
+    language = gr.Dropdown(label="Language", choices=list(options.keys()), value="English")
+    task = gr.Dropdown(label="Task", choices=list(options['English'].keys()))
+    message = gr.Textbox(label="Input Text")
+    max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
+    temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
+    top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
+    output = gr.Textbox(label="Output")
+
+    language.change(update_tasks, inputs=language, outputs=task)
+    message.submit(respond, inputs=[message, language, task, system_message, max_tokens, temperature, top_p], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()