Update app.py

app.py

@@ -5,9 +5,9 @@ from transformers import pipeline
 model_name = "TDN-M/tknv1"  # Thay thế bằng tên người dùng và tên repository của bạn
 pipe = pipeline("text-generation", model=model_name, max_new_tokens=2048)
 
-def chatbot(input_text):
+def chatbot(message, history):
     # Tạo tin nhắn từ người dùng
-    messages = [{"role": "user", "content": input_text}]
+    messages = [{"role": "user", "content": message}]
     
     # Sử dụng mô hình để tạo phản hồi
     response = pipe(messages)
@@ -15,13 +15,13 @@ def chatbot(input_text):
     # Trả về phản hồi từ mô hình
     return response[0]['generated_text']
 
-# Tạo ứng dụng Gradio
-iface = gr.Interface(
+# Tạo ứng dụng Gradio với giao diện chatbot hiện đại
+iface = gr.ChatInterface(
     fn=chatbot,
-    inputs="text",
-    outputs="text",
-    title="Chatbot",
-    description="chatbot sử dụng tiếng Việt",
+    title="Chatbot sử dụng tiếng Việt",
+    description="Một chatbot sử dụng mô hình đã được huấn luyện bằng tiếng Việt.",
+    theme="default",  # Chọn theme cho giao diện
+    css=".gradio-container {background-color: #f0f0f0;}"  # Thêm CSS tùy chỉnh
 )
 
 # Chạy ứng dụng