Spaces:
Sleeping
Sleeping
zhiweili
commited on
Commit
•
d7867bc
1
Parent(s):
c1f0cb1
change pipeline
Browse files- app_haircolor.py +1 -1
app_haircolor.py
CHANGED
@@ -23,7 +23,7 @@ DEFAULT_CATEGORY = "hair"
|
|
23 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
24 |
|
25 |
def create_demo() -> gr.Blocks:
|
26 |
-
from inversion_run_realvxl
|
27 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
28 |
model_path = 'realesr-general-x4v3.pth'
|
29 |
half = True if torch.cuda.is_available() else False
|
|
|
23 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
24 |
|
25 |
def create_demo() -> gr.Blocks:
|
26 |
+
from inversion_run_realvxl import run as realvxl_run
|
27 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
28 |
model_path = 'realesr-general-x4v3.pth'
|
29 |
half = True if torch.cuda.is_available() else False
|