Update app.py

app.py

@@ -2,19 +2,21 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
+# Tải mô hình chỉ 1 lần duy nhất
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-small")
 
-def chatbot(message):
-    input_ids = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
-    output = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
-    reply = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
-    return "🐱 Dr. Ask: " + reply
+# Hàm chat đơn giản
+def chatbot(msg):
+    input_ids = tokenizer.encode(msg + tokenizer.eos_token, return_tensors='pt')
+    output_ids = model.generate(input_ids, max_length=100, pad_token_id=tokenizer.eos_token_id)
+    reply = tokenizer.decode(output_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+    return f"🐱 Dr. Ask: {reply}"
 
 demo = gr.Interface(fn=chatbot,
-                    inputs=gr.Textbox(label="Bạn muốn hỏi gì?"),
-                    outputs=gr.Textbox(label="Trả lời từ Dr. Ask"),
-                    title="🐱 Dr. Ask Chatbot",
-                    description="Tư vấn AI thân thiện — powered by DialoGPT-small")
+                    inputs=gr.Textbox(label="Bạn hỏi gì nè?"),
+                    outputs=gr.Textbox(label="Dr. Ask trả lời"),
+                    title="🐱 Dr. Ask AI",
+                    theme="default")
 
 demo.launch()