Update app.py

app.py

@@ -403,6 +403,9 @@ def apply_mask(d, b):
     edge = []
     mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
     dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
+    mask = cv2.dilate(mask, dilation)
+    b = b*2+1
+    dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
     dmask = cv2.dilate(mask, dilation)
 
     d["background"][mask==255] = (255,255,255,255)