add blur to mask and changed mask process

.gitignore

@@ -4,4 +4,5 @@
 IP-Adapter
 ./models
 ./sdxl_models
-DPT/weights
+DPT/weights
+.workspace

processing/mask.py

@@ -1,17 +1,18 @@
 import cv2
 import numpy as np
 import torch
-from PIL import Image
+from PIL import Image, ImageFilter
 from rembg import remove
 from torchvision.transforms import Compose
 
 from DPT.dpt.transforms import PrepareForNet, NormalizeImage, Resize
 
 
-def create_mask(image):
-    rm_bg = remove(image)
-    target_mask = rm_bg.convert("RGB").point(lambda x: 0 if x < 1 else 255).convert('L').convert('RGB')
-    return target_mask.resize((1024, 1024))
+def create_mask(image, blur=True):
+    rm_bg = remove(image, post_process_mask=True, only_mask=True)
+    if blur:
+        rm_bg = rm_bg.filter(ImageFilter.GaussianBlur(3))
+    return rm_bg.resize((1024, 1024))
 
 
 def create_depth_map(image, model):

processing/mask.pyi

@@ -3,9 +3,11 @@ import torch
 from PIL.Image import Image
 
 
-def create_mask(image: Image) -> Image:
+def create_mask(image: Image, blur: bool = ...) -> Image:
     """
         Create a mask from the input image.
+        :param image: The input image.
+        :param blur: Whether to blur the mask. Default is True.
     """
     ...