Update app.py

app.py

@@ -34,6 +34,8 @@ def respond(message, language, task, max_tokens):
     )
     # 解码回复
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # 去除输入部分
+    response = response[len(user_message):].strip()
     return response
 
 # 更新任务选项的函数
@@ -42,7 +44,7 @@ def update_tasks(language):
 
 # 创建Gradio接口
 with gr.Blocks() as demo:
-    gr.Markdown("# Chatbot Interface")
+    gr.Markdown("# Open-domain Information Extraction Large Language Models Demo")
     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")