minor updates

app.py

@@ -108,8 +108,8 @@ def plot_activations(activation_1, activation_2, origin='lower'):
         fig.colorbar(ims[i], cax=cax, orientation='vertical')
         
     # ax0.set_title('Input', fontsize=titles)
-    ax1.set_title('Activation 1', fontsize=titles)
-    ax2.set_title('Activation 2', fontsize=titles)
+    ax1.set_title('Early Activation', fontsize=titles)
+    ax2.set_title('Late Activation', fontsize=titles)
     
     return fig
 
@@ -165,7 +165,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     fig1 = plot_activations(activation_1, activation_2, origin=origin)
     
     # plot raw activations
-    output, input_image, activation_1, activation_2 = get_activations(model, image, sub_mean=False)
+    _, _, activation_1, activation_2 = get_activations(model, image, sub_mean=False)
     activation_1 = normalize_array(activation_1)
     activation_2 = normalize_array(activation_2)
     fig2 = plot_activations(activation_1, activation_2, origin=origin)