fix(api): enable tiling when fixing faces after upscaling

ssube · Jan 16, 2023 · ba3eff5 · ba3eff5
1 parent b37f8cd
commit ba3eff5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py
@@ -76,7 +76,7 @@ def to(self, device):
         return self
 
 
-def make_resrgan(model_path):
+def make_resrgan(model_path, tile=0):
     model_path = path.join(model_path, resrgan_name + '.pth')
     if not path.isfile(model_path):
         for url in resrgan_url:
@@ -114,7 +114,7 @@ def upscale_resrgan(source_image: Image, model_path: str, faces=True) -> Image:
     output, _ = upsampler.enhance(image, outscale=outscale)
 
     if faces:
-        output = upscale_gfpgan(output, upsampler)
+        output = upscale_gfpgan(output, make_resrgan(model_path, 512))
 
     return Image.fromarray(output, 'RGB')