bug fix

app.py

@@ -130,7 +130,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     
     ax0, ax1, ax2 = axs[0], axs[1], axs[2]
     
-    im0 = ax1.imshow(input_image, cmap=cmap,
+    im0 = ax0.imshow(input_image, cmap=cmap,
                      origin=origin)    
     im1 = ax1.imshow(activation_1, cmap=cmap,
                      origin=origin)
@@ -145,8 +145,8 @@ def predict_and_analyze(model_name, num_channels, dim, image):
         fig.colorbar(ims[i], cax=cax, orientation='vertical')
         
     ax0.set_title('Input', fontsize=titles)
-    ax1.set_title('First Activation', fontsize=titles)
-    ax2.set_title('Second Activation', fontsize=titles)
+    ax1.set_title('Activation 1', fontsize=titles)
+    ax2.set_title('Activation 2', fontsize=titles)
     
     print("Sending to Hugging Face")