Update app.py

app.py

@@ -6,11 +6,12 @@ tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-def chat_doctor(message, history):
+def ask_doctor(message, history):
     prompt = f"ผู้ป่วย: {message}\nหมอ:"
-    result = generator(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)[0]["generated_text"]
-    return result.split("หมอ:")[-1].strip()
+    result = generator(prompt, max_new_tokens=256, temperature=0.7)[0]["generated_text"]
+    answer = result.split("หมอ:")[-1].strip()
+    return answer + "\n\n⚠️ ข้อมูลนี้เพื่อการศึกษาเท่านั้น ไม่แทนการวินิจฉัยจากแพทย์จริง"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🩺 หมอ AI ภาษาไทย\n> ให้คำแนะนำสุขภาพด้วยโมเดลภาษาไทย (สำหรับการศึกษา)")
-    gr.ChatInterface(fn=chat_doctor).launch()
+    gr.Markdown("# หมอ AI (ภาษาไทย)")
+    gr.ChatInterface(fn=ask_doctor, title="แชตกับหมอ AI 🤖").launch()