Update handler.py

handler.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Any
 from ultralytics import YOLO
+import os
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -7,8 +8,8 @@ class EndpointHandler():
         # pseudo:
         # self.model= load_model(path)
         # yolov8_model_name = 'yolov8_2023-07-19_yolov8m.pt'
-        print(path)
-        self.model = YOLO('./yolov8_2023-07-19_yolov8m.pt')    
+        print(os.listdir(path))
+        self.model = YOLO(os.path.join(path, 'yolov8_2023-07-19_yolov8m.pt'))    
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """