Updated app.py

app.py

@@ -15,16 +15,13 @@ OUTPUT_DIR = "/tmp/output"
 # Hugging Face Model Repository
 REPO_ID = "FrancescoLR/FLAMeS-model"  # Replace with your actual model repository ID
 
-# Function to download the model files from Hugging Face Model Hub
+# Function to download the Dataset004_WML folder
 def download_model():
     if not os.path.exists(DATASET_DIR):
         os.makedirs(DATASET_DIR, exist_ok=True)
         print("Downloading Dataset004_WML...")
-        hf_hub_download(repo_id=REPO_ID, filename="Dataset004_WML.zip", cache_dir=MODEL_DIR)
-        # Unzip the dataset into the correct location
-        subprocess.run(["unzip", "-o", os.path.join(MODEL_DIR, "Dataset004_WML.zip"), "-d", DATASET_DIR])
-        os.remove(os.path.join(MODEL_DIR, "Dataset004_WML.zip"))
-        print("Dataset004_WML downloaded and extracted.")
+        hf_hub_download(repo_id=REPO_ID, filename="Dataset004_WML", cache_dir=MODEL_DIR)
+        print("Dataset004_WML downloaded.")
 
 # Function to run nnUNet inference
 def run_nnunet_predict(nifti_file):