add progress bar
Browse files
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
|