Update handler.py

handler.py

@@ -19,9 +19,9 @@ class EndpointHandler:
         # print("Current working directory:", current_directory)
         
         url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm"
-        gdown.download(url, './best.pt', quiet=False)
+        gdown.download(url, 'best.pt', quiet=False)
         
-        self.model = YOLO("./best.pt")
+        self.model = YOLO("best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         inputs = data.get("inputs")
         print("in call")
@@ -33,7 +33,7 @@ class EndpointHandler:
         print("is path")
         print(path)
         # path = "http://10.10.2.100/cam-lo.jpg"
-        
+        model = self.model
 ###########################  Load Image  #################################       
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")
@@ -57,7 +57,7 @@ class EndpointHandler:
         # t-shirts-detector/1
         # mainmodel/2
         #result = self.CLIENT.infer(path, model_id="mainmodel/2")
-        result = self.model(img)
+        result = model(img)
         #annotated_frame = result[0].plot()
         detections = result[0].boxes
         #print(result[0].boxes.xyxy)