Update app.py

app.py

@@ -554,8 +554,10 @@ def apply_mask(d,b):
     global edge
 
     edge = []
-    mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY) - 128
-    mask[mask<0] = 0
+    mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
+    mask[mask<128] = 128
+    mask = mask - 128
+    print(mask[mask>0])
 
     mask, bgdModel, fgdModel = cv2.grabCut(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB), mask, None,None,None,15, cv2.GC_INIT_WITH_MASK)
     mask = np.where((mask==2)|(mask==0),1,0).astype('uint8')