Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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=
|
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"]
|