#!/bin/bash
echo "🚀 Downloading brain tumor segmentation models..."

# Create models directory
mkdir -p models

# Option 1: MONAI BraTS model (recommended)
curl -L -o models/monai_brats_model.pth \
    "https://catalog.ngc.nvidia.com/api/v1/models/nvidia/monaitoolkit/monai_brats_mri_segmentation/files/models/model.ts"

# Option 2: Attention U-Net from HuggingFace
curl -L -o models/attention_unet.atommic \
    "https://huggingface.co/wdika/SEG_UNet3D_BraTS2023AdultGlioma/resolve/main/SEG_UNet3D_BraTS2023AdultGlioma.atommic"

echo "✅ Models downloaded successfully!"