Update app.py

app.py

@@ -22,13 +22,13 @@ classes = ('plane', 'car', 'bird', 'cat', 'deer',
 model_layer_names = ["0", "1", "2", "3"]
 
 def get_layer(layer_name):
-    if layer_name == "0":
+    if layer_name == 0:
         return [model.prep[-1]]
-    elif layer_name == "1":
+    elif layer_name == 1:
         return [model.layer1[-1]]
-    elif layer_name == "2":
+    elif layer_name == 2:
         return [model.layer2[-1]]
-    elif layer_name == "3":
+    elif layer_name == 3:
         return [model.layer3[-1]]
     else:
         return None