Update handler.py

handler.py

@@ -18,11 +18,12 @@ class EndpointHandler:
         # current_directory = os.getcwd()
         # print("Current working directory:", current_directory)
         
+        
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm/view?usp=sharing"
         gdown.download(url, 'best.pt', quiet=False)
         
-        self.model = YOLO("best.pt")
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        model = YOLO("best.pt")
         inputs = data.get("inputs")
         print("in call")
         
@@ -33,7 +34,7 @@ class EndpointHandler:
         print("is path")
         print(path)
         # path = "http://10.10.2.100/cam-lo.jpg"
-        model = self.model
+        # model = self.model
 ###########################  Load Image  #################################       
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")