Update app.py

app.py

@@ -1,22 +1,27 @@
 import gradio as gr
 from transformers import pipeline
 
-model_name = "ZombitX64/Thai-sentiment-e5"  # เปลี่ยนเป็นชื่อโมเดลของคุณ
+model_name = "ZombitX64/Thai-sentiment-e5"
 nlp = pipeline("sentiment-analysis", model=model_name)
 
-# สร้าง dict แม็ป label เป็นเลข
 label_map = {
-    "negative": 0,
-    "neutral": 1,
-    "positive": 2
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
+}
+label_name_map = {
+    "LABEL_0": "negative",
+    "LABEL_1": "neutral",
+    "LABEL_2": "positive"
 }
 
 def analyze_text(text):
     result = nlp(text)[0]
-    label = result['label'].lower()  # แปลงเป็น lowercase เผื่อโมเดลให้แบบนี้
+    label = result['label']
     score = result['score']
-    code = label_map.get(label, -1)  # ถ้า label ไม่อยู่ใน dict ให้เป็น -1
-    return f"ผลวิเคราะห์: {label} (รหัส: {code}) ความมั่นใจ {score:.2f}"
+    code = label_map.get(label, -1)
+    label_name = label_name_map.get(label, label)
+    return f"ผลวิเคราะห์: {label_name} (รหัส: {code}) ความมั่นใจ {score:.2f}"
 
 demo = gr.Interface(
     fn=analyze_text,