chaojiemao commited on
Commit
0356673
·
verified ·
1 Parent(s): 0c9fdf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)