Update app.py

app.py

@@ -1,12 +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)
 
+label_map = {
+    "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]
-    return f"ผลวิเคราะห์: {result['label']} (ความมั่นใจ {result['score']:.2f})"
+    label = result['label']
+    score = result['score']
+    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,