zhiweili commited on
Commit
e049791
1 Parent(s): a4d9ce1

change gpu time

Browse files
Files changed (1) hide show
  1. app_haircolor.py +2 -2
app_haircolor.py CHANGED
@@ -9,6 +9,7 @@ from segment_utils import(
9
  restore_result,
10
  )
11
  from enhance_utils import enhance_image
 
12
 
13
 
14
  DEFAULT_SRC_PROMPT = "a woman, with hair"
@@ -19,9 +20,8 @@ DEFAULT_CATEGORY = "hair"
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
 
21
  def create_demo() -> gr.Blocks:
22
- from inversion_run_adapter import run as adapter_run
23
 
24
- @spaces.GPU(duration=10)
25
  def image_to_image(
26
  input_image: Image,
27
  input_image_prompt: str,
 
9
  restore_result,
10
  )
11
  from enhance_utils import enhance_image
12
+ from inversion_run_adapter import run as adapter_run
13
 
14
 
15
  DEFAULT_SRC_PROMPT = "a woman, with hair"
 
20
  device = "cuda" if torch.cuda.is_available() else "cpu"
21
 
22
  def create_demo() -> gr.Blocks:
 
23
 
24
+ @spaces.GPU(duration=15)
25
  def image_to_image(
26
  input_image: Image,
27
  input_image_prompt: str,