bug fix

app.py

@@ -333,7 +333,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     input_fig = plot_input(input_image, origin=origin)
     
     # plot mean subtracted activations
-    fig1 = plot_activations(activation_1, activation_2, model_name, origin=origin)
+    fig1 = plot_activations(activation_1, activation_2, origin=origin)
     
     # plot raw activations
     _, _, activation_1, activation_2 = get_activations(model, image, model_name, 
@@ -341,7 +341,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
                                                        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)
+    fig2 = plot_activations(activation_1, activation_2, origin=origin)
     
     print("Sending to Hugging Face")