apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): remove unused upsampler arg

This commit is contained in:
Sean Sube 2023-02-06 08:56:34 -06:00
parent 24ab52bf0d
commit 564cfc1279
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/chain/correct_gfpgan.py
View File

@ -1,16 +1,13 @@
from logging import getLogger
from os import path
from typing import Optional
import numpy as np
from gfpgan import GFPGANer
from PIL import Image
from realesrgan import RealESRGANer
from ..device_pool import JobContext
from ..params import DeviceParams, ImageParams, StageParams, UpscaleParams
from ..utils import ServerContext, run_gc
from .upscale_resrgan import load_resrgan
logger = getLogger(__name__)
@ -19,9 +16,7 @@ last_pipeline_instance = None
last_pipeline_params = None
def load_gfpgan(
ctx: ServerContext, upscale: UpscaleParams, _device: DeviceParams
):
def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams, _device: DeviceParams):
global last_pipeline_instance
global last_pipeline_params
@ -56,7 +51,6 @@ def correct_gfpgan(
source_image: Image.Image,
*,
upscale: UpscaleParams,
upsampler: Optional[RealESRGANer] = None,
**kwargs,
) -> Image.Image:
if upscale.correction_model is None:
@ -65,7 +59,7 @@ def correct_gfpgan(
logger.info("correcting faces with GFPGAN model: %s", upscale.correction_model)
device = job.get_device()
gfpgan = load_gfpgan(server, upscale, device, upsampler=upsampler)
gfpgan = load_gfpgan(server, upscale, device)
output = np.array(source_image)
_, _, output = gfpgan.enhance(