fixing size error

handler.py

@@ -256,10 +256,11 @@ class EndpointHandler:
             image = image.convert("RGB")
 
         w, h = image.size
+        # Use full-resolution inference for maximum accuracy; YOLO expects [h, w]
         if image_size is None:
-            imgsz = {"h": h, "w": w}
+            imgsz = [h, w]
         else:
-            imgsz = [image_size.get("h", h), image_size.get("w", w)]
+            imgsz = [int(image_size.get("h", h)), int(image_size.get("w", w))]
 
         out = self.yolo.predict(
             image,
@@ -267,6 +268,7 @@ class EndpointHandler:
             conf=bbox_threshold,
             iou=iou_threshold or 0.7,
             verbose=False,
+            augment=True,  # enable TTA for accuracy (scales/flips)
         )[0]
         if out.boxes is None:
             raise RuntimeError(