This is a project to fine-tune a pretrained ViT to identify pneumonia in chest x-ray images.
The goal of this project is to take the Pneumonia Chest X-Ray dataset, upload it to the Hugging Face Hub, create a classifier to detect pneumonia, and finally upload the model to the Hugging Face Hub.
.
├── README.md
├── models <- compiled model .pkl or HDFS or .pb format
├── config <- any configuration files
├── data
│ ├── processed <- data after all preprocessing has been done
│ └── raw <- original unmodified data acting as source of truth and provenance
├── notebooks <- jupyter notebooks for exploratory analysis and explanation
├── reports <- generated project artefacts eg. visualisations or tables
│ ├── figures
│ └── raw <- original unmodified data acting as source of truth and
└── src
├── examples <- images used as examples in the HuggingFace Space
├── models <- scripts for the prediction of the images
├── visualization <- scripts to visualize the atttention vectors
├── app.py <- script to execute the gradio app
└── requirements.txt <- text file containing the required libraries