fix(api): make stage-specific source image optional in GFPGAN stage (#…

…233)
ssube · Mar 11, 2023 · 6bdaa4a · 6bdaa4a
1 parent da3d95f
commit 6bdaa4a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)