Update app.py

app.py

@@ -27,7 +27,8 @@ scaler = MinMaxScaler()
 for j in range(6):
     layer=conv_layers[layer_ind]
     weights=layer.get_weights()[0][:,:,0,j]
-    norm_weights = scaler.fit_transform(weights)
+    #norm_weights = scaler.fit_transform(weights)
+    norm_weights=weights
     plt.subplot(2,3,j+1)
     plt.imshow(norm_weights,cmap='gray')
     plt.title(f"Conv Layer")
@@ -80,7 +81,7 @@ 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], cmap='gray')
+            plt.imshow(fm[:, :, i])
             plt.title(f"Feature Map {i+1}")
             plt.axis('off')