Update app.py

app.py

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