From dab2cc7752d4a6c062729643e87b1dcdbcc62940 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 16 Jan 2023 17:51:32 -0600
Subject: [PATCH] lint(api): split up face correction and upscaling models

---
 api/onnx_web/serve.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index 43fce6ad..ceed0f39 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -98,6 +98,9 @@ mask_filters = {
 # TODO: load from model_path
 upscale_models = [
     'RealESRGAN_x4plus',
+]
+
+face_models = [
     'GFPGANv1.3',
 ]
 
@@ -186,7 +189,7 @@ def upscale_from_request() -> UpscaleParams:
         scale=scale,
         outscale=outscale,
         faces=faces,
-        face_model=upscale_models[1],
+        face_model=face_models[0],
         platform='onnx',
         denoise=denoise,
     )