asafaya/bert-base-arabic Model usage

models/space_classify.py

@@ -10,16 +10,16 @@ def classifyA(text):
     """
     classification = pipeline(
         task="text-classification",
-        model="Hashuz/AS_MentalQAU",
+        model="asafaya/bert-base-arabic",
         return_all_scores=True
     )
 
     result = []
     # Define mapping dictionary
     mapping = {
-        'info': 'تقديم معلومة',
-        'guid': 'توجيه أو ارشاد',
-        'support': 'دعم نفسي'
+        'LABEL_0': 'تقديم معلومة',
+        'LABEL_1': 'توجيه أو ارشاد',
+        'LABEL_2': 'دعم نفسي'
     }
 
     output = classification(text)