README.md

metadata

title: ERA S12
emoji: 🌖
colorFrom: green
colorTo: purple
sdk: gradio
sdk_version: 3.39.0
app_file: app.py
pinned: false
license: mit

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

-> We trained CIFAR10 dataset using Custom ResNet model by using pytorch lightning.

-> requirements.txt file contains necessary packages to install.

-> custom_resnet.py file contains model architecture.

-> model_pth.ckpt contains trained model checkpoints (weights).

-> 10 example images like cat_1.jpg, cat_2.jpg,..

--> app.py contains gradio code. By using gradio here implemented by selecting input images or examples output display the gradcam image and prediction and top k classes. I tried with multiple gradcam and misclassified i am getting errors.