fffiloni commited on
Commit
ac1c798
1 Parent(s): 58c884a

add progress bars

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -86,7 +86,7 @@ def cuda_memory_cleanup():
86
  gc.collect()
87
 
88
  @spaces.GPU(duration=90)
89
- def infer(frame1_path, frame2_path):
90
 
91
  seed = 42
92
  num_inference_steps = 10
 
86
  gc.collect()
87
 
88
  @spaces.GPU(duration=90)
89
+ def infer(frame1_path, frame2_path, progress=gr.Progress(track_tqdm=True)):
90
 
91
  seed = 42
92
  num_inference_steps = 10