This repo contains files for the cancer detection project. We used PyTorch-Lightning framework to build an image classification model. To track hyperparameters and metrics, we utilized Comet ML. We also used Gradio to create a cancer detection app. To examine this project, you can check our app on HuggingFace.
📌 Medium blog post: Simplifying the Image Classification Workflow with Lightning & Comet ML