Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -445,7 +445,7 @@ class DemoUI(object):
|
|
445 |
|
446 |
@spaces.GPU(duration=60)
|
447 |
def run_example(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
|
448 |
-
output_h, output_w, seed):
|
449 |
model_path = self.task_model[task_type]["MODEL_PATH"]
|
450 |
|
451 |
step = self.pipe.input.get("sample_steps", 20)
|
|
|
445 |
|
446 |
@spaces.GPU(duration=60)
|
447 |
def run_example(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
|
448 |
+
output_h, output_w, seed, progress=gr.Progress(track_tqdm=True)):
|
449 |
model_path = self.task_model[task_type]["MODEL_PATH"]
|
450 |
|
451 |
step = self.pipe.input.get("sample_steps", 20)
|