added top code for number of classes

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/app.cpython-37.pyc
+__pycache__/custom_resnet.cpython-37.pyc

app.py

@@ -36,8 +36,9 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1, num_top_c
     # Sort confidences dictionary in descending order of values and take top num_top_classes
     sorted_confidences = {k: v for k, v in sorted(confidences.items(), key=lambda item: item[1], reverse=True)}
     top_classes = list(sorted_confidences.keys())[:num_top_classes]
+    top = dict((k,v) for k, v in sorted_confidences.items() if k in top_classes)
 
-    return top_classes, visualization
+    return top, visualization
 
 title = "CIFAR10 trained on ResNet18 Model with GradCAM"
 description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"