Fixes #351: weights_path for "task_id == 8"

wasserth · Aug 11, 2024 · 6b1781f · 6b1781f
1 parent e36752a
commit 6b1781f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py
@@ -281,7 +281,7 @@ def download_pretrained_weights(task_id):
         # WEIGHTS_URL = url + "/static/totalseg_v2/Dataset503_cardiac_motion.zip"
         WEIGHTS_URL = url + "/v2.0.0-weights/Dataset503_cardiac_motion.zip"
     elif task_id == 8:
-        weights_path = config_dir / "Task008_HepaticVessel"
+        weights_path = config_dir / "Dataset008_HepaticVessel"
         WEIGHTS_URL = url + "/v2.4.0-weights/Dataset008_HepaticVessel.zip"
 
     # Commercial models