From 6bdaa4a4edd734b58e1f463a5397aa2fe4a8006e Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 10 Mar 2023 18:15:31 -0600
Subject: [PATCH] fix(api): make stage-specific source image optional in GFPGAN
 stage (#233)

---
 api/onnx_web/chain/correct_gfpgan.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py
index 2cff2e18..6041bee2 100644
--- a/api/onnx_web/chain/correct_gfpgan.py
+++ b/api/onnx_web/chain/correct_gfpgan.py
@@ -1,5 +1,6 @@
 from logging import getLogger
 from os import path
+from typing import Optional
 
 import numpy as np
 from PIL import Image
@@ -54,7 +55,7 @@ def correct_gfpgan(
     source: Image.Image,
     *,
     upscale: UpscaleParams,
-    stage_source: Image.Image,
+    stage_source: Optional[Image.Image] = None,
     **kwargs,
 ) -> Image.Image:
     upscale = upscale.with_args(**kwargs)