auto update

Author: taylor-arnold

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()
 

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)