Update app.py

app.py

@@ -47,7 +47,8 @@ def remove_noise(noisy_image, password):
         sigma_est = np.mean(estimate_sigma(image_float))
 
     denoised_image = denoise_nl_means(image_float, h=1.15 * sigma_est, fast_mode=True,
-                                      patch_size=5, patch_distance=6, multichannel=False)
+                                      patch_size=5, patch_distance=6, channel_axis=None)
+
     denoised_image = (255 * denoised_image).astype(np.uint8)  # Scale back to 0-255
     return Image.fromarray(denoised_image)