MakeNTU-v2024
/

Image-preprocessor

Model card Files Files and versions

oscarfu0501 commited on May 5, 2024

Commit

20fa134

·

verified ·

1 Parent(s): 3906f8f

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -21,9 +21,9 @@ class EndpointHandler:
     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)
-        model = YOLO("best.pt")
         inputs = data.get("inputs")
         print("in call")

     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)
+        model = YOLO("./best.pt")
         inputs = data.get("inputs")
         print("in call")