Update smoothening

app.py

@@ -184,6 +184,10 @@ def smooth_and_denoise(mask):
     :param mask: The binary mask of the signature.
     :return: Processed mask.
     """
+    # Ensure the mask is a 2D array
+    if mask.ndim > 2:
+        mask = mask[..., 0]
+
     # Apply Gaussian Blurring for smoothing
     smoothed_mask = cv2.GaussianBlur(mask, (5, 5), 0)
 
@@ -195,6 +199,7 @@ def smooth_and_denoise(mask):
                                      patch_size=5, patch_distance=3, channel_axis=None)
     return denoised_mask
 
+
     
 def predict(net,  inputs_val, shapes_val, hypar, device):
     '''