from modules import shared class FaceRestoration: def name(self): return "None" def restore(self, np_image): return np_image def restore_faces(np_image, p=None): 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, p)