Update app.py

app.py

@@ -586,8 +586,6 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
     if len(edge) == 0 or params["fnum"] != frame_selected or params["l"] != l:
       if len(edge) > 0:
           d["background"] = cv2.imread(depths[frame_selected]).astype(np.uint8)
-          mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8) + 128
-          mask[mask==128] = 0
 
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)
       bg[bg==255] = 0
@@ -595,7 +593,10 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
       params["fnum"] = frame_selected
       params["l"] = l
 
+      mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8) + 128
+      mask[mask==128] = 0
       d["layers"][0] = cv2.cvtColor(mask, cv2.COLOR_GRAY2RGBA)
+        
       edge = bg.copy()
     else:
       bg = edge.copy()