ArchCoder commited on
Commit
bed938d
·
verified ·
1 Parent(s): cad0957

Create download_model.sh

Browse files
Files changed (1) hide show
  1. 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!"