MakeNTU-v2024
/

Image-preprocessor

Model card Files Files and versions

oscarfu0501 commited on May 5, 2024

Commit

82559f3

·

verified ·

1 Parent(s): 23eb631

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -19,9 +19,9 @@ class EndpointHandler:
         # print("Current working directory:", current_directory)
         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]]:
         inputs = data.get("inputs")
         print("in call")

         # print("Current working directory:", current_directory)
         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]]:
         inputs = data.get("inputs")
         print("in call")