Update handler.py

handler.py

@@ -3,6 +3,8 @@ from typing import Dict, List, Any
 from ultralytics import YOLO
 import json
 from urllib.request import urlopen
+from PIL import Image
+from io import BytesIO
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -42,8 +44,9 @@ class EndpointHandler():
         image_path = event["inputs"]
 
         try:
+            image = Image.Image()._new(Image.open(BytesIO(image)))
             #image = urlopen(image_path).read()
-            predictions = self.predict_objects(image_path)
+            predictions = self.predict_objects(imag)
             return {
                 "statusCode": 200,
                 "body": json.dumps(predictions),