MakeNTU-v2024
/

Image-preprocessor

Model card Files Files and versions Community

oscarfu0501 commited on May 5, 2024

Commit

eeaeb0e

·

verified ·

1 Parent(s): f64f1c0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -18,7 +18,7 @@ class EndpointHandler:
         # current_directory = os.getcwd()
         # print("Current working directory:", current_directory)
-        url = "https://drive.google.com/uc?id=1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm"
         gdown.download(url, './best.pt', quiet=False)
         self.model = YOLO("./best.pt")

         # current_directory = os.getcwd()
         # 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")