zhiweili commited on
Commit
d7867bc
1 Parent(s): c1f0cb1

change pipeline

Browse files
Files changed (1) hide show
  1. 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.py 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
 
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