Add application file

app.py

@@ -2,27 +2,23 @@ import gradio as gr
 from transformers import pipeline
 
 pipe = pipeline(
-    "zero-shot-classification",
-    model="backzone1/FI-Chat-Classify"  
+    "text-classification",
+    model="backzone1/FI-Chat-Classify",  # โมเดลที่คุณเทรนเองและอัปโหลด
+    tokenizer="backzone1/FI-Chat-Classify"  # ให้ใช้ tokenizer จากโมเดลเดียวกัน
 )
 
 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})"
+    result = pipe(msg)[0]
+    label = result["label"]
+    score = result["score"]
+    return f"ผลลัพธ์: {label} (ความมั่นใจ {score:.2f})"
 
 demo = gr.Interface(
     fn=classify_message,
     inputs=gr.Textbox(lines=5, label="พิมพ์ข้อความจาก LINE"),
     outputs=gr.Textbox(label="ผลลัพธ์"),
     title="จำแนกข้อความว่าเป็นงานหรือไม่",
-    description="ใช้โมเดล zero-shot classification ที่เทรนแล้วเพื่อวิเคราะห์ข้อความ"
+    description="ใช้โมเดลที่ฝึกเองเพื่อจำแนกข้อความ"
 )
 
 demo.launch()