Update app.py

app.py

@@ -360,6 +360,9 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
     else:
         gradient = cv2.imread(f[2]).astype(np.uint8)
 
+    if len(d["layers"]) == 1:
+        d["layers"][1] = d["layers"][0].copy()
+
     #for i in range(n):
     #    im[(im >= i*255/n) & (im < (i+1)*255/n)] = i*255/(n-1)
     indices = np.arange(0,256)   # List of all colors