fffiloni commited on
Commit
32d0b90
·
verified ·
1 Parent(s): 08d577c

show tqdm progress

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def get_video_fps(video_path):
109
  def load_examples(ref_image_in, ref_video_in):
110
  return "./examples/mimicmotion_result1_example.mp4"
111
 
112
- def infer(ref_image_in, ref_video_in, num_inference_steps, guidance_scale, output_frames_per_second, seed, checkpoint_version):
113
  # check if 'outputs' dir exists and empty it if necessary
114
  check_outputs_folder('./outputs')
115
 
 
109
  def load_examples(ref_image_in, ref_video_in):
110
  return "./examples/mimicmotion_result1_example.mp4"
111
 
112
+ def infer(ref_image_in, ref_video_in, num_inference_steps, guidance_scale, output_frames_per_second, seed, checkpoint_version, progress=gr.Progress(track_tqdm=True)):
113
  # check if 'outputs' dir exists and empty it if necessary
114
  check_outputs_folder('./outputs')
115