Update app.py

app.py

@@ -349,13 +349,13 @@ def draw_mask(v, d, evt: gr.EventData):
     fgdModel = np.zeros((1,65), np.float64)
     rect = (x, y, w, h)
     mask = cv2.grabCut(rgb, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT)
-    newmask = mask[0]
-    newmask = np.where((newmask==2)|(newmask==0),1,0).astype('uint8')
-    print(rgb)
+    mask0 = cv2.cvtColor(mask[0], cv2.COLOR_RGB2RGBA)
+    newmask = np.where((mask0==2)|(mask0==0),1,0).astype('uint8')
+    print(mask0)
     print(newmask)
     img = cv2.cvtColor(rgb, cv2.COLOR_RGB2RGBA)*newmask[:,:,np.newaxis]
 
-    d["background"] = img
+    d["layers"][0]= mask0
     d["composite"] = img
     
     return gr.ImageEditor(value=d)