auto update

distant-viewing · Feb 9, 2023 · 9af8383 · 9af8383
1 parent 73ec9aa
commit 9af8383
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dvt/detect.py b/dvt/detect.py
@@ -116,7 +116,7 @@ class AnnoDetect:
     """
     def __init__(self, model_path=None):
         if not model_path:
-            model_path = _download_file("dvt_detect.pt")
+            model_path = _download_file("dvt_detect.pt", "1.0.1")
         self.model = torch.jit.load(model_path)
         self.model.eval()
 

diff --git a/dvt/utils.py b/dvt/utils.py
@@ -111,9 +111,12 @@ def save_image(image_path: str, img: np.ndarray) -> None:
 
 def _download_file(
     url: str,
-    basename="https://github.com/distant-viewing/dvt/releases/download/0.0.1/"
+    basename="https://github.com/distant-viewing/dvt/releases/download/": str,
+    version="0.0.1": str
 ) -> str:
 
+    basename = basename + version
+
     if basename:
         url = os.path.join(basename, url)