Update app.py

app.py

@@ -1,17 +1,15 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-model_id = "iapp/chinda-qwen3-4b"
+model_id = "pythainlp/KhanomTanLLM-1B"  # โมเดลฟรี รองรับภาษาไทย
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+model     = AutoModelForCausalLM.from_pretrained(model_id)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-def ask_doctor(message, history):
-    prompt = f"ผู้ป่วย: {message}\nหมอ:"
-    result = generator(prompt, max_new_tokens=256, temperature=0.7)[0]["generated_text"]
-    answer = result.split("หมอ:")[-1].strip()
-    return answer + "\n\n⚠️ ข้อมูลนี้เพื่อการศึกษาเท่านั้น ไม่แทนการวินิจฉัยจากแพทย์จริง"
+def chatbot_fn(message, history):
+    prompt = f"มนุษย์: {message}\nAI:"
+    out = generator(prompt, max_new_tokens=200, temperature=0.7)[0]["generated_text"]
+    answer = out.split("AI:")[-1].strip()
+    return answer
 
-with gr.Blocks() as demo:
-    gr.Markdown("# หมอ AI (ภาษาไทย)")
-    gr.ChatInterface(fn=ask_doctor, title="แชตกับหมอ AI 🤖").launch()
+gr.ChatInterface(fn=chatbot_fn, title="AI ภาษาไทย (KhanomTan)").launch()