fffiloni commited on
Commit
251568b
·
verified ·
1 Parent(s): 7223d1e

add progress bar

Browse files
Files changed (1) hide show
  1. webgui.py +1 -1
webgui.py CHANGED
@@ -289,7 +289,7 @@ with gr.Blocks() as demo:
289
  </div>
290
  """)
291
 
292
- def generate_video(uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device):
293
 
294
  final_output_path = process_video(
295
  uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device
 
289
  </div>
290
  """)
291
 
292
+ def generate_video(uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device, progress=gr.Progress(track_tqdm=True)):
293
 
294
  final_output_path = process_video(
295
  uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device