Update app.py

app.py

@@ -404,9 +404,10 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
       params["l"] = l
       params["c"] = c
       params["k"] = k
+
+      d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
     else:
-      bg = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
-      mask = cv2.inRange(bg, 250, 255)
+      mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
 
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
@@ -423,7 +424,6 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
     grad = cv2.addWeighted(grad_x, 0.5, grad_y, 0.5, 0)
     mask[cv2.inRange(grad, 1, 255)>0] = 1
     
-    d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
     d["layers"][1][mask>0] = (0,0,0,255)
     
     return gr.ImageEditor(value=d)