Update app.py

app.py

@@ -346,6 +346,7 @@ def draw_mask(v, d, evt: gr.EventData):
     bgdModel = np.zeros((1,65),np.float64)
     fgdModel = np.zeros((1,65),np.float64)
 
+    mask = []
     mask[newmask == 0] = 0
     mask[newmask == 255] = 1
     mask, bgdModel, fgdModel = cv2.grabCut(img,mask,None,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_MASK)
@@ -355,7 +356,7 @@ def draw_mask(v, d, evt: gr.EventData):
     d["layers"][0] = 255-img
     d["composite"][0] = 255-img
     
-    return d
+    return gr.ImageEditor(value=d)
 
 
 css = """
@@ -406,7 +407,7 @@ with gr.Blocks(css=css) as demo:
             output_frame = gr.Gallery(label="Frames", preview=True, columns=8192, type="numpy")
             output_switch = gr.Checkbox(label="Show depths")
             output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
-            output_mask = gr.ImageEditor(layers=True, sources=None, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(default_size=0, colors=['black', 'darkgray', 'gray', 'lightgray', 'white']), elem_id="image_edit")
+            output_mask = gr.ImageEditor(layers=True, sources=None, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(default_size=0, colors=['black', '#505050', '#a0a0a0', 'white']), elem_id="image_edit")
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')