yaya36095 commited on
Commit
217d46e
·
verified ·
1 Parent(s): 41fab92

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -4
handler.py CHANGED
@@ -61,7 +61,7 @@ class EndpointHandler:
61
  print(f"خطأ في فك الترميز: {e}")
62
  raise
63
 
64
- def __call__(self, data: Any) -> Dict[str, Any]:
65
  print(f"استدعاء __call__ مع نوع البيانات: {type(data)}")
66
 
67
  img = None
@@ -77,7 +77,7 @@ class EndpointHandler:
77
 
78
  if img is None:
79
  print("لم يتم العثور على صورة صالحة")
80
- return {"label": "error", "score": 1.0}
81
 
82
  print("تحويل الصورة إلى مدخلات الموديل")
83
  inputs = self.processor(images=img, return_tensors="pt").to(self.device)
@@ -97,8 +97,8 @@ class EndpointHandler:
97
  results.sort(key=lambda x: x["score"], reverse=True)
98
  best = results[0]
99
  print(f"أفضل نتيجة: {best}")
100
- return best
101
 
102
  except Exception as e:
103
  print(f"حدث استثناء: {e}")
104
- return {"label": "error", "score": 1.0}
 
61
  print(f"خطأ في فك الترميز: {e}")
62
  raise
63
 
64
+ def __call__(self, data: Any) -> List[Dict[str, Any]]:
65
  print(f"استدعاء __call__ مع نوع البيانات: {type(data)}")
66
 
67
  img = None
 
77
 
78
  if img is None:
79
  print("لم يتم العثور على صورة صالحة")
80
+ return [{"label": "error", "score": 1.0}]
81
 
82
  print("تحويل الصورة إلى مدخلات الموديل")
83
  inputs = self.processor(images=img, return_tensors="pt").to(self.device)
 
97
  results.sort(key=lambda x: x["score"], reverse=True)
98
  best = results[0]
99
  print(f"أفضل نتيجة: {best}")
100
+ return [best]
101
 
102
  except Exception as e:
103
  print(f"حدث استثناء: {e}")
104
+ return [{"label": "error", "score": 1.0}]