This registry contains the model for making pneumonia predictions and was prepared for Bielefeld University Deep Learning course homework.
The code used for this implementation mostly comes from here:
https://github.com/nateraw/huggingpics it was a ready pipeline for model fine-tuning with huggingface and PyTorch Lightning for another dataset.