Update app.py

app.py

@@ -20,6 +20,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"]
 
 def get_layer(layer_name):
     if layer_name == "0":
@@ -89,7 +90,7 @@ demo = gr.Interface(
         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.Dropdown(layer_name, value="layer3_x", label="Which layer for Gradcam"),
+        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"),
     ],
     outputs = [