Update app.py

app.py

@@ -88,11 +88,11 @@ def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0
 demo = gr.Interface(
     inference,
     inputs = [
-        gr.Image(width=256,height=256,label="input image"),
+        gr.Image(width=256,height=256,label="Input image"),
         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.Checkbox(True, label="Show GradCAM Image"),
         gr.Dropdown(model_layer_names, value="3", label="Which layer for Gradcam"),
-        gr.Slider(0,1,value=0.5,label="Overall opacity of the overlay"),
+        gr.Slider(0, 1, value=0.5,label="Overall opacity of the overlay"),
     ],
     outputs = [
         gr.Label(label="Class", container=True, show_label= True),
@@ -101,7 +101,7 @@ demo = gr.Interface(
     ],
     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, "1", -1],
+    examples = [["cat.jpg", True, 10, "1", -1],
                 ["dog.jpg", 4,  False, "1", -1]]
 )