minor updates

app.py

@@ -187,6 +187,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     W = int(dim)
 
     print("Loading data")
+    print(image)
     image = np.load(image.name, allow_pickle=True)
     image = image.astype(np.float32)
     
@@ -209,7 +210,9 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     print("Model loaded")
     
     print("Looking at activations")
-    output, input_image, activation_1, activation_2 = get_activations(model, image, model_name, sub_mean=True)
+    output, input_image, activation_1, activation_2 = get_activations(model, image, model_name, 
+                                                                      channel=input_channel,
+                                                                      sub_mean=True)
     print("Activations and predictions finished")
     
     if output[0] < output[1]:
@@ -235,7 +238,9 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     fig1 = plot_activations(activation_1, activation_2, model_name, origin=origin)
     
     # plot raw activations
-    _, _, activation_1, activation_2 = get_activations(model, image, model_name, sub_mean=False)
+    _, _, activation_1, activation_2 = get_activations(model, image, model_name, 
+                                                       channel=input_channel,
+                                                       sub_mean=False)
     activation_1 = normalize_array(activation_1)
     activation_2 = normalize_array(activation_2)
     fig2 = plot_activations(activation_1, activation_2, model_name, origin=origin)