Update app.py

app.py

@@ -91,16 +91,17 @@ demo = gr.Interface(
         gr.Number(value=3, maximum=10, minimum=1,step=1.0, precision=0,label="Number of classes to display"),
         gr.Checkbox(True,label="Show GradCAM Image"),
         gr.Dropdown(model_layer_names, value="layer3_x", label="Which layer for Gradcam"),
-        gr.Slider(0,1,value=0.5,label="Overall opacity of the overelay"),
+        gr.Slider(0,1,value=0.5,label="Overall opacity of the overlay"),
     ],
     outputs = [
-        gr.Image(width= 256, height=256,label="Output"),
-        gr.Label(label="Confidences", container=True, show_label= True)        
+        gr.Label(label="Class", container=True, show_label= True),
+        gr.Image(width= 256, height=256,label="Output Image"),
+        gr.Label(label="Confidences", container=True, show_label= True),
     ],
     title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
     description = " A simple gradio inference to infer on resnet18 model",
     examples = [["cat.jpg", 10, True, "layer3_x", -1],
-                ["dog.jpg", 4,  False, "layer3_x", -1]]
+                ["dog.jpg", 4,  False, "layer2_x", -1]]
 )
 
 if __name__ == "__main__":