Update app.py

app.py

@@ -335,6 +335,10 @@ def switch_rows(v):
         print(frames[0])
         return frames
 
+def draw_mask(v, evt: gr.EventData):
+    print(evt)
+    return v
+
 
 css = """
 #img-display-container {
@@ -385,6 +389,8 @@ with gr.Blocks(css=css) as demo:
             output_switch = gr.Checkbox(label="Show depths")
             output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
             output_mask = gr.ImageEditor(layers=False, sources=None, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(colors=['black', 'darkgray', 'gray', 'lightgray', 'white']))
+            output_mask.input(fn=draw_mask, inputs=[output_mask], outputs=[output_mask])
+            output_mask.apply(fn=draw_mask, inputs=[output_mask], outputs=[output_mask])
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')