Update app.py

app.py

@@ -13,13 +13,16 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def respond(message, history):
     # 將當前訊息與歷史訊息合併
-    input_text = message if not history else history[-1][1] + " " + message
+    input_text = message if not history else history[-1]["content"] + " " + message
     # 獲取模型的回應
     response = pipe(input_text, max_length=100, num_return_sequences=1)
-    return response[0]['generated_text'], history + [(message, response[0]['generated_text'])]
+    reply = response[0]['generated_text']
+    
+    # 返回新的消息格式
+    return reply, history + [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
 
 # 設定 Gradio 的聊天界面
-demo = gr.ChatInterface(fn=respond)
+demo = gr.ChatInterface(fn=respond, title="Chat with Qwen 2.5", description="與 HelloSun/Qwen2.5-0.5B-Instruct-openvino 聊天！", type='messages')
 
 if __name__ == "__main__":
     demo.launch()