Update handler.py

handler.py

@@ -5,10 +5,7 @@ import os
 class EndpointHandler():
     def __init__(self, path=""):
         # Preload all the elements you are going to need at inference.
-        # pseudo:
-        # self.model= load_model(path)
-        # yolov8_model_name = '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]]: