fffiloni commited on
Commit
b094879
·
verified ·
1 Parent(s): fa60e88

track tqdm in Geadio UI

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -39,7 +39,7 @@ hf_hub_download(
39
  )
40
 
41
  @spaces.GPU()
42
- def generate(control_sequence, ref_image):
43
  control_image = control_sequence # "data_test/sample4.mp4"
44
  ref_image = ref_image # "data_test/sample4.png"
45
  unique_id = str(uuid.uuid4())
 
39
  )
40
 
41
  @spaces.GPU()
42
+ def generate(control_sequence, ref_image, progress=gr.Progress(track_tqdm=True)):
43
  control_image = control_sequence # "data_test/sample4.mp4"
44
  ref_image = ref_image # "data_test/sample4.png"
45
  unique_id = str(uuid.uuid4())