Update app.py

app.py

@@ -3,7 +3,7 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import gradio as gr
 
-
+# موديل التحليل الفارسي
 model_name = "HooshvareLab/bert-fa-base-uncased-sentiment-snappfood"
 
 # تحميل الموديل والتوكنيزر
@@ -16,17 +16,21 @@ sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=token
 # دالة التحليل
 def analyze_sentiment(text):
     results = sentiment_analyzer(text)[0]
+
+    # احتمال تكون التسميات LABEL_0 / LABEL_1 / LABEL_2
+    labels_map = {
+        "LABEL_0": "منفی (سلبي)",
+        "LABEL_1": "خنثی (محايد)",
+        "LABEL_2": "مثبت (إيجابي)"
+    }
+
     output = ""
     for r in results:
         label = r["label"]
+        label_fa = labels_map.get(label, label)
         score = round(r["score"], 3)
-        if label.lower() == "positive":
-            label_fa = "مثبت (إيجابي)"
-        elif label.lower() == "negative":
-            label_fa = "منفی (سلبي)"
-        else:
-            label_fa = "خنثی (محايد)"
         output += f"{label_fa}: {score}\n"
+
     return output
 
 # واجهة Gradio