d0tpy commited on
Commit
62e3f73
·
verified ·
1 Parent(s): 27a87ea

Update image_enhancer.py

Browse files
Files changed (1) hide show
  1. image_enhancer.py +2 -2
image_enhancer.py CHANGED
@@ -36,7 +36,7 @@ class Enhancer:
36
  return
37
 
38
  model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=self.upscale)
39
- model_path = f'https://huggingface.co/dtarnow/UPscaler/resolve/main/RealESRGAN_x{self.upscale}plus.pth'
40
  self.bg_upsampler = RealESRGANer(
41
  scale=self.upscale,
42
  model_path=model_path,
@@ -51,7 +51,7 @@ class Enhancer:
51
  raise ValueError('CUDA is not available for RealESRGAN')
52
 
53
  model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=self.upscale)
54
- model_path = f'https://huggingface.co/dtarnow/UPscaler/resolve/main/RealESRGAN_x{self.upscale}plus.pth'
55
  self.realesrgan_enhancer = RealESRGANer(
56
  scale=self.upscale,
57
  model_path=model_path,
 
36
  return
37
 
38
  model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=self.upscale)
39
+ model_path = f'https://huggingface.co/dtarnow/UPscaler/resolve/main/RealESRGAN_x2plus.pth'
40
  self.bg_upsampler = RealESRGANer(
41
  scale=self.upscale,
42
  model_path=model_path,
 
51
  raise ValueError('CUDA is not available for RealESRGAN')
52
 
53
  model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=self.upscale)
54
+ model_path = f'https://huggingface.co/dtarnow/UPscaler/resolve/main/RealESRGAN_x2plus.pth'
55
  self.realesrgan_enhancer = RealESRGANer(
56
  scale=self.upscale,
57
  model_path=model_path,