Spaces:
Sleeping
Sleeping
zhiweili
commited on
Commit
•
e049791
1
Parent(s):
a4d9ce1
change gpu time
Browse files- 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=
|
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,
|