Update handler.py
Browse files- 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}]
|