chaojiemao commited on
Commit
d8af134
·
verified ·
1 Parent(s): 6def55d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -358,7 +358,7 @@ class DemoUI(object):
358
  err_msg = "Please provide the reference image or edited image."
359
  return None, None, None, False, err_msg
360
  return edit_image, edit_mask, ref_image, is_suc, err_msg
361
- @spaces.GPU(duration=60)
362
  def run_chat(
363
  prompt,
364
  ref_image,
@@ -444,7 +444,7 @@ class DemoUI(object):
444
  outputs=chat_outputs,
445
  queue=True)
446
 
447
- @spaces.GPU(duration=60)
448
  def run_example(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
449
  output_h, output_w, seed, progress=gr.Progress(track_tqdm=True)):
450
  model_path = self.task_model[task_type]["MODEL_PATH"]
 
358
  err_msg = "Please provide the reference image or edited image."
359
  return None, None, None, False, err_msg
360
  return edit_image, edit_mask, ref_image, is_suc, err_msg
361
+ @spaces.GPU(duration=80)
362
  def run_chat(
363
  prompt,
364
  ref_image,
 
444
  outputs=chat_outputs,
445
  queue=True)
446
 
447
+ @spaces.GPU(duration=80)
448
  def run_example(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
449
  output_h, output_w, seed, progress=gr.Progress(track_tqdm=True)):
450
  model_path = self.task_model[task_type]["MODEL_PATH"]