Update app.py

app.py

@@ -19,7 +19,7 @@ inv_normalize = transforms.Normalize(
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
            'dog', 'frog', 'horse', 'ship', 'truck')
 
-model_layer_names = ["prep", "layer1_x", "layer2_x", "layer3_x"]
+model_layer_names = ["0", "1", "2", "3"]
 
 def get_layer(layer_name):
     if layer_name == "0":
@@ -99,8 +99,8 @@ 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, "layer3_x", -1],
-                ["dog.jpg", 4,  False, "layer2_x", -1]]
+    examples = [["cat.jpg", 10, True, "3", -1],
+                ["dog.jpg", 4,  False, "2", -1]]
 )
 
 if __name__ == "__main__":