bug fix

app.py

@@ -126,6 +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()
+    
+    image = image.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()