Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ if not os.path.exists('CodeFormer.pth'):
|
|
33 |
model_us = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
34 |
model_us_path = 'realesr-general-x4v3.pth'
|
35 |
half = True if torch.cuda.is_available() else False
|
36 |
-
upsampler = RealESRGANer(scale=4,
|
37 |
|
38 |
os.makedirs('output', exist_ok=True)
|
39 |
|
|
|
33 |
model_us = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
34 |
model_us_path = 'realesr-general-x4v3.pth'
|
35 |
half = True if torch.cuda.is_available() else False
|
36 |
+
upsampler = RealESRGANer(scale=4, model_path=model_us_path, model=model_us, tile=0, tile_pad=10, pre_pad=0, half=half)
|
37 |
|
38 |
os.makedirs('output', exist_ok=True)
|
39 |
|