From d7b67d9b40e47ede766d3beb149b0c2b74651ece Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 3 Sep 2022 09:05:48 +0300
Subject: [PATCH] load GFPGAN once and keep it in memory for #53

---
 webui.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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]