bug fix

app.py

@@ -126,8 +126,8 @@ def get_activations(model, image: list, model_name: str,
         if i == max(activation_indices[model_name]):
             break
     output = model.model(image).detach().cpu().numpy()
-    output_1 = activation_indices[model_name][0].detach().cpu().numpy()
-    output_2 = activation_indices[model_name][1].detach().cpu().numpy()
+    output_1 = layer_outputs[activation_indices[model_name][0]].detach().cpu().numpy()
+    output_2 = layer_outputs[activation_indices[model_name][1]].detach().cpu().numpy()
     
     # get activations
     # output_1 = outputs[1]