hysts HF staff commited on
Commit
84c9342
1 Parent(s): 6877bee

Add progress bar

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -539,7 +539,14 @@ def add_tracking_points(
539
 
540
 
541
  @spaces.GPU
542
- def run(first_frame_path, last_frame_path, tracking_points, controlnet_cond_scale, motion_bucket_id):
 
 
 
 
 
 
 
543
  original_width, original_height = 512, 320 # TODO
544
 
545
  # load_image
 
539
 
540
 
541
  @spaces.GPU
542
+ def run(
543
+ first_frame_path,
544
+ last_frame_path,
545
+ tracking_points,
546
+ controlnet_cond_scale,
547
+ motion_bucket_id,
548
+ progress=gr.Progress(track_tqdm=True),
549
+ ):
550
  original_width, original_height = 512, 320 # TODO
551
 
552
  # load_image