Update app.py

app.py

@@ -170,7 +170,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
 
         global frame_selected
         global masks
-        masks = [gr.ImageEditor(interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(), layers=True)] * len(orig_frames)
+        masks = orig_frames
         return final_vid, final_zip, orig_frames, depth_frames, masks[frame_selected] #output_path
 
 def depth_edges_mask(depth):
@@ -363,7 +363,7 @@ with gr.Blocks(css=css) as demo:
             input_url.change(fn=loadurl, inputs=[input_url], outputs=[input_video])
             output_frame = gr.Gallery(label="Frame", type='numpy', preview=True, columns=8192)
             output_depth = gr.Gallery(label="Depth", type='numpy', preview=True, columns=8192, interactive=False)
-            output_mask = gr.ImageEditor(interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(), layers=True)
+            output_mask = gr.ImageEditor(interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(colors=['black', 'darkgray', 'gray', 'lightgray', 'white']), layers=True)
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')