Spaces:
Running
on
Zero
Running
on
Zero
add progress bars
Browse files- 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
|