Update app.py

app.py

@@ -52,7 +52,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(feature_maps[:, :, i], cmap='gray')
+        plt.imshow(fm[:, :, i], cmap='gray')
         plt.title(f"Feature Map {i}")
         plt.axis('off')