updated downloadable model default path

rajewsky-lab · Apr 19, 2024 · 69eefc9 · 69eefc9
1 parent c9fb92d
commit 69eefc9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/openst/preprocessing/image_preprocess.py b/openst/preprocessing/image_preprocess.py
@@ -25,7 +25,7 @@
  "HE_CUT_rajewsky"
 ]
 
-MODEL_DIR = pathlib.Path.home().joinpath(".CUT", "models")
+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "CUT", "models")
 _MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/CUT_models"
 
 def create_dataset(args, opt):

diff --git a/openst/segmentation/segment.py b/openst/segmentation/segment.py
@@ -32,7 +32,7 @@
 ]
 
 # adapted from cellpose
-MODEL_DIR = pathlib.Path.home().joinpath(".cellpose", "models")
+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "cellpose", "models")
 _MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/models"
 
 def cache_model_path(basename):