Update app.py

app.py

@@ -269,7 +269,7 @@ def get_mesh(image, depth, blur_data, loadall):
     #print(image[fnum][0])
     #print(depth["composite"])
 
-    depthc = cv2.cvtColor(depth["composite"], cv2.COLOR_RGBA2RGB)
+    depthc = cv2.cvtColor(depth["background"], cv2.COLOR_RGBA2RGB)
     blur_img = blur_image(image[fnum][0], depthc, blur_data)
     gdepth = rgb2gray(depthc)
     
@@ -429,6 +429,7 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
     mask[cv2.inRange(grad, 1, 255)>0] = 1
     
     d["layers"][1][mask>0] = (0,0,0,255)
+    d["background"][mask>0] = (0,0,0,255)
     
     return gr.ImageEditor(value=d)