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
|