Update app.py

app.py

@@ -6,7 +6,17 @@ clf = pipeline("text-classification", model="distilbert-base-uncased")
 
 def classify_text(text):
     result = clf(text)[0]
-    return f"Label: {result['label']} (Score: {round(result['score'], 2)})"
+    label = result['label']
+    score = round(result['score'], 2)
+
+    # เปลี่ยนชื่อ label ให้เข้าใจง่าย
+    if label == "LABEL_1":
+        label_name = "ความคิดเห็นเชิงบวก 😀"
+    else:
+        label_name = "ความคิดเห็นเชิงลบ 😞"
+
+    return f"{label_name} (ความมั่นใจ: {score})"
+
 
 demo = gr.Interface(fn=classify_text, inputs="text", outputs="text", title="Text Classifier")