Update handler.py

handler.py

@@ -2,6 +2,7 @@ from huggingface_hub import hf_hub_download
 from typing import Dict, List, Any
 from ultralytics import YOLO
 import json
+from urllib.request import urlopen
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -40,7 +41,8 @@ class EndpointHandler():
         image_path = event["inputs"]
 
         try:
-            predictions = self.predict_objects(image_path)
+            image = urlopen(image_path).read()
+            predictions = self.predict_objects(image)
             return {
                 "statusCode": 200,
                 "body": json.dumps(predictions),