SkalskiP commited on
Commit
84fb73b
β€’
1 Parent(s): 0ccb481

Added example usage for ProPainter feature

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -20,6 +20,9 @@ This is a demo for automatic removal of objects from videos using
20
  - [x] Automated inpainting using ProPainter
21
  - [ ] Automated ⚑ object masking using FastSAM + MetaCLIP
22
  """
 
 
 
23
 
24
  START_FRAME = 0
25
  END_FRAME = 10
@@ -149,6 +152,15 @@ with gr.Blocks() as demo:
149
  with gr.Column():
150
  masked_video_player = gr.Video(label="Masked video")
151
  painted_video_player = gr.Video(label="Painted video")
 
 
 
 
 
 
 
 
 
152
 
153
  submit_button.click(
154
  process,
 
20
  - [x] Automated inpainting using ProPainter
21
  - [ ] Automated ⚑ object masking using FastSAM + MetaCLIP
22
  """
23
+ EXAMPLES = [
24
+ ["https://media.roboflow.com/supervision/video-examples/ball-juggling.mp4", "person", 0.6]
25
+ ]
26
 
27
  START_FRAME = 0
28
  END_FRAME = 10
 
152
  with gr.Column():
153
  masked_video_player = gr.Video(label="Masked video")
154
  painted_video_player = gr.Video(label="Painted video")
155
+ with gr.Row():
156
+ gr.Examples(
157
+ examples=EXAMPLES,
158
+ fn=process,
159
+ inputs=[source_video_player, prompt_text, confidence_slider],
160
+ outputs=[masked_video_player, painted_video_player],
161
+ cache_examples=False,
162
+ run_on_click=True
163
+ )
164
 
165
  submit_button.click(
166
  process,