Update app.py

app.py

@@ -25,11 +25,11 @@ def get_layer(layer_name):
     if layer_name == "0":
         return [model.prep[-1]]
     elif layer_name == "1":
-        return [model.layer1_x[-1]]
+        return [model.layer1[-1]]
     elif layer_name == "2":
-        return [model.layer2_x[-1]]
+        return [model.layer2[-1]]
     elif layer_name == "3":
-        return [model.layer3_x[-1]]
+        return [model.layer3[-1]]
     else:
         return None