Spaces:
Running
Running
Update image_enhancer.py
Browse files- 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/
|
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/
|
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,
|