got tid of .model

app.py

@@ -88,11 +88,11 @@ def get_activations(model, image: list, model_name: str,
     
     
     layer_outputs = {}
-    for i in range(len(model.model.features)):
-        image = model.model.features[i](image)
+    for i in range(len(model.features)):
+        image = model.features[i](image)
         layer_outputs[i] = image
         print(i, layer_outputs[i].shape)
-    output = model.model(image).detach().cpu().numpy()
+    output = model(image).detach().cpu().numpy()
     output_1 = activation_indices[model_name].detach().cpu().numpy()
     output_2 = activation_indices[model_name].detach().cpu().numpy()