Update handler.py

handler.py

@@ -7,15 +7,15 @@ from ultralytics import YOLO
 
 
 class EndpointHandler:
-    def __init__(self, key):  # pass api key to model
-        pass
+    def __init__(self, path='.'):  # pass api key to model
+        self.model = YOLO("./best.pt")
         
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         inputs = data.get("inputs")
         isurl = inputs.get("isurl")
         path = inputs.get("path")
         
-        model = YOLO("MakeNTU-v2024/Image-preprocessor/best.pt")
+        model = self.model
 ###########################  Load Image  #################################       
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")