Update app.py

app.py

@@ -341,18 +341,15 @@ def draw_mask(v, d, evt: gr.EventData):
     w = int(vv[2])
     h = int(vv[3])
 
-    d_copy = d["background"].copy()
-    rgb = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB)
-    d["background"] = cv2.applyColorMap(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY), cv2.COLORMAP_HSV)
-    mask = np.zeros(rgb.shape[:2], np.uint8)
+    mask = np.zeros(d["background"].shape[:2], np.uint8)
     bgdModel = np.zeros((1,65), np.float64)
     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)
+    mask = cv2.grabCut(d["background"], mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT)
     print(mask)
     mask0 = mask[0]
     newmask = np.where((mask0==2)|(mask0==0),1,0).astype('uint8')
-    d["composite"] = cv2.cvtColor(rgb*newmask[:,:,np.newaxis], cv2.COLOR_RGB2RGBA)
+    d["composite"] = rgb*newmask[:,:,np.newaxis]
     
     return gr.ImageEditor(value=d)