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
|