revert this

CBICA · Oct 9, 2024 · 092a19a · 092a19a
1 parent 4bc1473
commit 092a19a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DLICV/__main__.py b/DLICV/__main__.py
@@ -255,8 +255,8 @@ def main() -> None:
         print("DLICV model not found, downloading...")
 
         from huggingface_hub import snapshot_download
-        # local_src = Path(__file__).parent
-        snapshot_download(repo_id="nichart/DLICV")
+        local_src = Path(__file__).parent
+        snapshot_download(repo_id="nichart/DLICV", local_dir=local_src)
         print("DLICV model has been successfully downloaded!")
     else:
         print("Loading the model...")