Update handler.py

handler.py

@@ -2,8 +2,8 @@ from huggingface_hub import hf_hub_download
 from typing import Dict, List, Any
 from ultralytics import YOLO
 import json
-from urllib.request import urlopen
-from PIL import Image
+import urllib.request
+import cv2
 from io import BytesIO
 
 class EndpointHandler():
@@ -42,12 +42,16 @@ class EndpointHandler():
         image_path = event["inputs"]
 
         try:
-            data=urlopen(image_path).read()
-            image = Image.open(BytesIO(data))
-            image_size = image.size
-            image = image.resize([1280,960])
-            predictions = self.predict_objects(image, [image_size[0]/1280,image_size[1]/960])
-            
+            with urllib.request.urlopen(image_path) as response:
+                image_content = np.asarray(bytearray(response.read()), dtype=np.uint8)
+            image = cv2.imdecode(image_content, cv2.IMREAD_COLOR)
+            image_size = image.shape
+            if image.shape[0]>image.shape[0]:
+                x, y = 1280, 960
+            else:
+                y, x = 1280, 960
+            image = cv2.resize(image, (x, y))
+            predictions = predict_objects(self, image, [image_size[0]/x,image_size[1]/y])
             return {
                 "statusCode": 200,
                 "body": json.dumps(predictions),