Update app.py

app.py

@@ -2,14 +2,7 @@ import streamlit as st
 from transformers import pipeline
 
 # تحميل النموذج
-classifier = pipeline("text-classification", model="distilbert-base-uncased")
-
-# قائمة العلامات
-labels = {
-    "LABEL_0": "shopping",
-    "LABEL_1": "gaming",
-    "LABEL_2": "streaming"
-}
+classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
 
 # عنوان التطبيق
 st.title("Text Classification App")
@@ -20,9 +13,14 @@ text = st.text_area("Enter your text here:")
 # زر التصنيف
 if st.button("Classify"):
     if text:
-        result = classifier(text)
-        label = result[0]['label']
-        mapped_label = labels.get(label, "Unknown")
-        st.write(f"Classification Result: {mapped_label}")
+        # تحديد الفئات التي تريد تصنيف النص إليها
+        categories = ["shopping", "gaming", "streaming"]
+        result = classifier(text, categories)
+        
+        # الحصول على الفئة الأكثر احتمالاً
+        best_category = result['labels'][0]
+        score = round(result['scores'][0], 2)
+        
+        st.write(f"Classification Result: {best_category} (Confidence: {score})")
     else:
         st.warning("Please enter some text!")