Update app.py

app.py

@@ -474,8 +474,10 @@ def apply_mask(d, b):
     edge = []
     mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
     mask[mask<255] = 0
+    b = b*2+1
     dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
     mask = cv2.dilate(mask, dilation)
+    mask_b = cv2.GaussianBlur(mask, (b,b), 0)
     b = b*2+1
     dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
     dmask = cv2.dilate(mask, dilation)
@@ -494,7 +496,7 @@ def apply_mask(d, b):
             cv2.imwrite(frames[k], frame)
         
     frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
-    frame[:, :, 3] = 255 - dmask_b
+    frame[:, :, 3] = 255 - mask_b
     cv2.imwrite(frames[frame_selected], frame)
     
     masks[frame_selected] = d["background"]