Updated app.py

app.py

@@ -19,9 +19,10 @@ REPO_ID = "FrancescoLR/FLAMeS-model"  # Replace with your actual model repositor
 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", cache_dir=MODEL_DIR)
-        print("Dataset004_WML downloaded.")
+        print("Downloading Dataset004_WML.zip...")
+        zip_path = hf_hub_download(repo_id=REPO_ID, filename="Dataset004_WML.zip", cache_dir=MODEL_DIR)
+        subprocess.run(["unzip", "-o", zip_path, "-d", MODEL_DIR])
+        print("Dataset004_WML downloaded and extracted.")
 
 # Function to run nnUNet inference
 def run_nnunet_predict(nifti_file):