Update README.md

README.md

@@ -10,4 +10,20 @@ pinned: false
 license: apache-2.0
 ---
 
+# IMAGE CLASSIFICATION with GradCAM
+
+A simple Gradio interface to visualize the output of a CNN trained on CIFAR10 dataset with GradCAM and Misclassified images. 
+The architecture is inspired from David Page’s (myrtle.ai) DAWNBench winning model archiecture. 
+Please input the image and select the number of top predictions to display - you will see the top predictions and their corresponding confidence scores. 
+You can also select whether to show GradCAM for the particular image (utilizes the gradients of the classification score with respect to the final convolutional feature map, to identify the parts of an input image that most impact the classification score). 
+You need to select the model layer where the gradients need to be plugged from - this affects how much of the image is used to compute the GradCAM. You can also select whether to show misclassified images - these are the images that the model misclassified. 
+Some examples are provided in the examples tab.
+
+## CIFAR-10 Dataset
+
+The CIFAR-10 dataset consists of 60000 32x32 colour images in 10 classes, with 6000 images per class. There are 50000 training images and 10000 test images.
+
+The dataset is divided into five training batches and one test batch, each with 10000 images. The test batch contains exactly 1000 randomly-selected images from each class. The training batches contain the remaining images in random order, but some training batches may contain more images from one class than another. Between them, the training batches contain exactly 5000 images from each class.
+
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference