Update app.py

app.py

@@ -615,10 +615,10 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
     d["layers"][0][sel==0] = (0,0,0,0)
 
 
-    mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY) - 128
-    mask[mask<0] = 0
-    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 = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
+    mask[mask==0] = 128
+    print(mask[mask>0]-128)
+    mask, bgdModel, fgdModel = cv2.grabCut(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB), mask-128, None,None,None,15, cv2.GC_INIT_WITH_MASK)
     mask = np.where((mask==2)|(mask==0),1,0).astype('uint8')
 
     frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)