Upload folder using huggingface_hub

README.md

@@ -0,0 +1,8 @@
+---
+language_bcp47:
+- en
+- vi-VN
+pipeline_tag: object-detection
+library_name: ultralytics
+license: mit
+---

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e7d536e4c3e994bcff3fc39e6659e94caca910b18b631aad29b1eeb2dd3d349
+size 19182483

handler.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict
+import io, base64, requests
+from PIL import Image
+from ultralytics import YOLO
+
+class EndpointHandler:
+    def __init__(self, path: str = ""):
+        self.model = YOLO(f"{path}/best.pt")
+
+    def _load_image(self, data: Dict) -> Image.Image:
+        if "inputs" in data:  # HF UI/SDK đôi khi gửi trường "inputs"
+            data = data["inputs"]
+
+        if isinstance(data, dict) and data.get("image_url"):
+            img_bytes = requests.get(data["image_url"], timeout=15).content
+            return Image.open(io.BytesIO(img_bytes)).convert("RGB")
+
+        if isinstance(data, dict) and data.get("image_base64"):
+            img_bytes = base64.b64decode(data["image_base64"])
+            return Image.open(io.BytesIO(img_bytes)).convert("RGB")
+
+        if isinstance(data, (bytes, bytearray)):
+            return Image.open(io.BytesIO(data)).convert("RGB")
+
+        raise ValueError("No image provided. Use 'image_url' or 'image_base64' or raw bytes.")
+
+    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+        conf = float(data.get("conf", 0.5)) if isinstance(data, dict) else 0.5
+        img = self._load_image(data)
+        res = self.model.predict(img, conf=conf, verbose=False)
+        r = res[0]
+        boxes_out = []
+        if r.boxes is not None:
+            for b in r.boxes:
+                xyxy = b.xyxy[0].tolist()
+                confv = float(b.conf[0])
+                cls = int(b.cls[0])
+                boxes_out.append({"xyxy": xyxy, "conf": confv, "cls": cls})
+        return {
+            "num_detections": len(boxes_out),
+            "boxes": boxes_out,
+            "max_confidence": max([b["conf"] for b in boxes_out], default=0.0),
+            "is_diseased": len(boxes_out) > 0
+        }

requirements.txt

@@ -0,0 +1,4 @@
+ultralytics==8.3.179
+torch==2.8.0
+pillow
+numpy