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 = [1, 2, 3]
+model_layer_names = ["1", "2", "3"]
 
 def get_layer(layer_name):
     print(layer_name)