Spaces:
Runtime error
Runtime error
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)
|