From 6a9b33c848281cb02f38764e4f91ef767f5e3edd Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Wed, 7 Sep 2022 12:32:28 +0300
Subject: codeformer support

---
 modules/face_restoration.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 modules/face_restoration.py

(limited to 'modules/face_restoration.py')

diff --git a/modules/face_restoration.py b/modules/face_restoration.py
new file mode 100644
index 00000000..4ae53d21
--- /dev/null
+++ b/modules/face_restoration.py
@@ -0,0 +1,19 @@
+from modules import shared
+
+
+class FaceRestoration:
+    def name(self):
+        return "None"
+
+    def restore(self, np_image):
+        return np_image
+
+
+def restore_faces(np_image):
+    face_restorers = [x for x in shared.face_restorers if x.name() == shared.opts.face_restoration_model or shared.opts.face_restoration_model is None]
+    if len(face_restorers) == 0:
+        return np_image
+
+    face_restorer = face_restorers[0]
+
+    return face_restorer.restore(np_image)
-- 
cgit v1.2.1