Spaces:
Sleeping
Sleeping
Update app_enhance.py
Browse files- app_enhance.py +1 -1
app_enhance.py
CHANGED
|
@@ -36,7 +36,7 @@ if not os.path.exists('realesr-general-x4v3.pth'):
|
|
| 36 |
|
| 37 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
| 38 |
model_path = 'realesr-general-x4v3.pth'
|
| 39 |
-
half = True if torch.cuda.is_available() else
|
| 40 |
upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
| 41 |
|
| 42 |
|
|
|
|
| 36 |
|
| 37 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
| 38 |
model_path = 'realesr-general-x4v3.pth'
|
| 39 |
+
half = True if torch.cuda.is_available() else "cpu"
|
| 40 |
upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
| 41 |
|
| 42 |
|