Added smooth and denoising with changes

app.py

@@ -171,23 +171,24 @@ def inference(image):
     image_path = image
     
     image_tensor, orig_size = load_image(image_path, hypar) 
-    mask = predict(net, image_tensor, orig_size, hypar, device)
+    original_mask = predict(net, image_tensor, orig_size, hypar, device)
     
-    # Process the mask without cropping
-    # processed_mask = smooth_and_denoise(mask)
+    # Process the original mask with smoothing and denoising
+    processed_mask = smooth_and_denoise(original_mask)
 
     # Convert processed mask to PIL image
-    pil_mask = Image.fromarray(mask).convert('L')
+    pil_processed_mask = Image.fromarray(processed_mask).convert('L')
     im_rgb = Image.open(image).convert("RGB")
 
     im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
     
-    # Apply mask to images
+    # Apply processed mask to images
     im_rgba = im_rgb.copy()
-    im_rgba.putalpha(pil_mask)
-    im_dark.putalpha(pil_mask)
+    im_rgba.putalpha(pil_processed_mask)
+    im_dark.putalpha(pil_processed_mask)
     
-    return [im_rgba, pil_mask, im_dark]
+    return [im_rgba, pil_processed_mask, im_dark]
+
 
 title = "Mysign.id - Signature Background removal based on DIS"
 description = "ML Model based on ECCV2022/dis-background-removal specifically made for removing background from signatures."