Add application file

app.py

@@ -1,34 +1,28 @@
 import gradio as gr
 from transformers import pipeline
 
-# โหลดโมเดล
 pipe = pipeline(
     "zero-shot-classification",
     model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
 )
 
-# ฟังก์ชันประมวลผลข้อความ
-def extract_fields(msg):
-    prompt = f"""กรุณาสรุปข้อมูลจากข้อความด้านล่าง แล้วแสดงผลเป็นรายการ
-- วันที่ขึ้นของ
-- วันที่ส่งของ
-- ต้นทาง
-- ปลายทาง
+def classify_message(msg):
+    labels = ["เป็นงาน", "ไม่ใช่งาน"]
+    result = pipe(
+        msg,
+        candidate_labels=labels,
+        hypothesis_template="ข้อความนี้เป็น {}"
+    )
+    best = result["labels"][0]
+    score = result["scores"][0]
+    return f"ผลลัพธ์: {best} (ความมั่นใจ {score:.2f})"
 
-Message:
-{msg}
-[/INST]
-"""
-    result = pipe(prompt, max_new_tokens=300, do_sample=False, temperature=0.3)
-    return result[0]["generated_text"]
-
-# UI ด้วย Gradio
 demo = gr.Interface(
-    fn=extract_fields,
-    inputs=gr.Textbox(lines=10, label="พิมพ์ข้อความจาก LINE"),
-    outputs=gr.Textbox(label="ข้อมูลที่สรุปได้"),
-    title="สรุปข้อมูลจากข้อความ LINE",
-    description="ระบบจะดึงข้อมูล วันที่ขึ้นของ, วันที่ส่งของ, ต้นทาง, ปลายทาง จากข้อความที่ได้รับ"
+    fn=classify_message,
+    inputs=gr.Textbox(lines=5, label="พิมพ์ข้อความจาก LINE"),
+    outputs=gr.Textbox(label="ผลลัพธ์"),
+    title="จำแนกข้อความว่าเป็นงานหรือไม่",
+    description="ใช้โมเดล zero-shot classification เพื่อวิเคราะห์ข้อความ"
 )
 
 demo.launch()