Update app.py

app.py

@@ -1,19 +1,23 @@
-from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+import gradio as gr
+from transformers import pipeline
 
-# ใช้โมเดลภาษาไทยที่ถูก fine-tune sentiment แล้ว (ตัวอย่าง)
-model_name = "pangpond/best-wangchanberta-sentiment"  # โมเดลนี้ถูกเทรนมาแล้ว
-clf = pipeline("text-classification", model=model_name, tokenizer=model_name)
+# โหลดโมเดลที่ฝึกแล้ว หรือโมเดล pretrained
+clf = pipeline("text-classification", model="distilbert-base-uncased")
 
 def classify_text(text):
     result = clf(text)[0]
     label = result['label']
     score = round(result['score'], 2)
 
-    if label == "pos":
+    # เปลี่ยนชื่อ label ให้เข้าใจง่าย
+    if label == "LABEL_1":
         label_name = "ความคิดเห็นเชิงบวก 😀"
-    elif label == "neg":
-        label_name = "ความคิดเห็นเชิงลบ 😞"
     else:
-        label_name = "ความคิดเห็นเป็นกลาง 😐"
+        label_name = "ความคิดเห็นเชิงลบ 😞"
 
     return f"{label_name} (ความมั่นใจ: {score})"
+
+
+demo = gr.Interface(fn=classify_text, inputs="text", outputs="text", title="Text Classifier")
+
+demo.launch()