Update handler.py

handler.py

@@ -47,7 +47,9 @@ class EndpointHandler:
                 print("تم تحميل الأوزان بنجاح")
 
             self.model.to(self.device).eval()
-            self.labels = self.model.config.id2label
+
+            # ✅ تعديل مهم جداً: تحويل المفاتيح إلى نصوص
+            self.labels = {str(k): v for k, v in self.model.config.id2label.items()}
 
         except Exception as e:
             print(f"خطأ أثناء تهيئة النموذج: {e}")
@@ -88,7 +90,7 @@ class EndpointHandler:
 
             results = []
             for i, prob in enumerate(probs):
-                label = self.labels[str(i)]
+                label = self.labels[str(i)]  # 🔥 المفتاح أصبح مضمون هنا
                 results.append({
                     "label": label,
                     "score": round(prob.item(), 4)