diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py index 963de7e9f..c3bc829d2 100644 --- 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