aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--webui.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/webui.py b/webui.py
index 503cc1e4..66e64a8e 100644
--- a/webui.py
+++ b/webui.py
@@ -179,8 +179,16 @@ def gfpgan_model_path():
return found[0]
+loaded_gfpgan_model = None
+
+
def gfpgan():
- return GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+ global loaded_gfpgan_model
+
+ if loaded_gfpgan_model is None:
+ loaded_gfpgan_model = GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+
+ return loaded_gfpgan_model
def gfpgan_fix_faces(gfpgan_model, np_image):
np_image_bgr = np_image[:, :, ::-1]