Spaces:
Sleeping
Sleeping
Create download_model.sh
Browse files- download_model.sh +15 -0
download_model.sh
ADDED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
echo "🚀 Downloading brain tumor segmentation models..."
|
3 |
+
|
4 |
+
# Create models directory
|
5 |
+
mkdir -p models
|
6 |
+
|
7 |
+
# Option 1: MONAI BraTS model (recommended)
|
8 |
+
curl -L -o models/monai_brats_model.pth \
|
9 |
+
"https://catalog.ngc.nvidia.com/api/v1/models/nvidia/monaitoolkit/monai_brats_mri_segmentation/files/models/model.ts"
|
10 |
+
|
11 |
+
# Option 2: Attention U-Net from HuggingFace
|
12 |
+
curl -L -o models/attention_unet.atommic \
|
13 |
+
"https://huggingface.co/wdika/SEG_UNet3D_BraTS2023AdultGlioma/resolve/main/SEG_UNet3D_BraTS2023AdultGlioma.atommic"
|
14 |
+
|
15 |
+
echo "✅ Models downloaded successfully!"
|