going back to just model

app.py

@@ -118,8 +118,8 @@ def get_activations(model, image: list, model_name: str,
     
     layer_outputs = {}
     temp_image = image
-    for i in range(len(model.model.features)):
-        temp_image = model.model.features[i](temp_image)
+    for i in range(len(model.features)):
+        temp_image = model.features[i](temp_image)
         if i in activation_indices[model_name]:
             layer_outputs[i] = image
             print(i, layer_outputs[i].shape)