Update app.py

app.py

@@ -63,7 +63,7 @@ if uploaded_img is not None:
     fig1 = plt.figure(figsize=(12, 4))
     for i in range(6):
         plt.subplot(2, 3, i + 1)
-        plt.imshow(fm[0,:, :, i], cmap='gray')
+        plt.imshow(fm[:, :, i], cmap='gray')
         plt.title(f"Feature Map {i+1}")
         plt.axis('off')