Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -19,9 +19,9 @@ class EndpointHandler:
|
|
19 |
# print("Current working directory:", current_directory)
|
20 |
|
21 |
url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm/view?usp=sharing"
|
22 |
-
gdown.download(url, '
|
23 |
|
24 |
-
self.model = YOLO("
|
25 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
26 |
inputs = data.get("inputs")
|
27 |
print("in call")
|
|
|
19 |
# print("Current working directory:", current_directory)
|
20 |
|
21 |
url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm/view?usp=sharing"
|
22 |
+
gdown.download(url, 'best.pt', quiet=False)
|
23 |
|
24 |
+
self.model = YOLO("best.pt")
|
25 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
26 |
inputs = data.get("inputs")
|
27 |
print("in call")
|