Update handler.py

handler.py

@@ -6,8 +6,8 @@ class EndpointHandler():
         # 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'
-        self.model = YOLO(yolov8_model_name)    
+        # yolov8_model_name = 'yolov8_2023-07-19_yolov8m.pt'
+        self.model = YOLO(path)    
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """