Update README.md

README.md

@@ -11,3 +11,15 @@ 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.