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!"
|