riyaarahim/mri-tumor-segmentation-api

MRI Brain Tumor Segmentation API

FastAPI service for automated brain tumor segmentation using a 3D U-Net trained on the BraTS 2021 dataset.

Endpoints

• —GET / — health check, confirms model is loaded
• —POST /predict — upload 4 MRI modalities (FLAIR, T1, T1ce, T2 as .nii.gz) to get tumor segmentation statistics

Visit /docs on the running Space for interactive API documentation.

Model

The trained model checkpoint is hosted separately at riyaarahim/mri-tumor-segmentation-unet and downloaded at container startup.

Architecture: 3D U-Net with skip connections, trained on BraTS 2021 (1,251 patients).