README.md

metadata

title: mri-classifier
emoji: 🌍
colorFrom: indigo
colorTo: indigo
sdk: gradio
sdk_version: 6.7.0
app_file: app.py
pinned: false
short_description: predictions on type of brain growth using fastai
license: mit
models:
  - fsattar/brain-tumor-classifier

brain_tumor_resnet34.ipynb is the notebook I used to train the model

It's based on resnet34 (relatively lightweight model good at identifying a lot of things)

Using the base model, we fine tune with our dataset to narrow the model's answer set to just our classes

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference