Update app.py

app.py

@@ -80,8 +80,8 @@ if uploaded_img is not None:
         fig1 = plt.figure(figsize=(100, 50))
         for i in range(120):
             plt.subplot(30, 4, i + 1)
-            plt.imshow(fm[:, :, i])
-            plt.title(f"Feature Map {i+1}",cmap="gray")
+            plt.imshow(fm[:, :, i],cmap="gray")
+            plt.title(f"Feature Map {i+1}")
             plt.axis('off')
 
     plt.tight_layout()