Update handler.py

handler.py

@@ -4,11 +4,23 @@ class EndpointHandler:
     def __init__(self, path=""):
         self.model = AutoModelForSequenceClassification.from_pretrained(
             path,
-            ignore_mismatched_sizes=True  # ضروري لتجاوز المشكلة
+            ignore_mismatched_sizes=True  # ضروري لتجاوز المشكلة اللي ظهرت في البداية
         )
         self.tokenizer = AutoTokenizer.from_pretrained(path)
         self.pipeline = pipeline("text-classification", model=self.model, tokenizer=self.tokenizer)
 
     def __call__(self, data):
-        inputs = data.get("inputs") if isinstance(data, dict) else data
-        return self.pipeline(inputs)
+        try:
+            # نحاول نجيب النص المدخل سواء من مفتاح inputs أو من القيمة نفسها
+            inputs = data.get("inputs", "") if isinstance(data, dict) else data
+
+            # تأكد إن المدخل نص مش قائمة أو فارغ
+            if not isinstance(inputs, str) or not inputs.strip():
+                return {"error": "Input must be a non-empty string."}
+
+            # شغل النموذج
+            result = self.pipeline(inputs)
+            return result
+
+        except Exception as e:
+            return {"error": str(e)}