From 6b1781f1612132bceb93ff18ca94ca919773927d Mon Sep 17 00:00:00 2001
From: Mario <6395323+mario-huang@users.noreply.github.com>
Date: Sun, 11 Aug 2024 12:30:03 +0800
Subject: [PATCH] Fixes #351: weights_path for "task_id == 8"

---
 totalsegmentator/libs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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