Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -13,6 +13,7 @@ import torch
 import torchvision
 import numpy as np
 import gradio as gr
+from gradio import Brush
 import tempfile
 import ffmpeg
 from PIL import Image
@@ -638,7 +639,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as iface:
         step2_title = gr.Markdown("---\n## Step2: Add masks", visible=False)
         with gr.Row(equal_height=True):
             with gr.Column(scale=2):
-                template_frame = gr.Image(type="numpy", label="Draw mask manually or click to add", tool="sketch", brush_radius=2, interactive=True, visible=False, elem_id="template_frame", elem_classes="image")
+                template_frame = gr.ImageEditor(type="numpy",label="Draw mask manually or click to add",interactive=True,visible=False,elem_id="template_frame",elem_classes="image",brush=gr.Brush(colors=["#FFFFFF"], default_color="#FFFFFF", default_size=2))
                 image_selection_slider = gr.Slider(minimum=1, maximum=100, step=1, value=1, label="Track start frame", visible=False)
                 track_pause_number_slider = gr.Slider(minimum=1, maximum=100, step=1, value=1, label="Track end frame", visible=False)
             with gr.Column(scale=2, elem_classes="jc_center"):