Update face_enhancer.py

face_enhancer.py

@@ -10,22 +10,12 @@ def gfpgan_runner(img, model):
     _, imgs, _ = model.enhance(img, paste_back=True, has_aligned=True)
     return imgs[0]
 
+
 def realesrgan_runner(img, model):
     img = model.predict(img)
     return img
 
-def realhatgan_runner(img, model):
-    img = model.predict(img)
-    return img
-        
-def restoreformer_runner(img, model):
-    img = model.predict(img)
-    return img
 
-def ultrasharp_runner(img, model):
-    img = model.predict(img)
-    return img
-    
 def codeformer_runner(img, model):
     img = model.enhance(img)
     return img
@@ -34,9 +24,6 @@ def codeformer_runner(img, model):
 supported_enhancers = {
     "CodeFormer": ("./assets/pretrained_models/codeformer.onnx", codeformer_runner),
     "GFPGAN": ("./assets/pretrained_models/GFPGANv1.4.pth", gfpgan_runner),
-    "REAL_HATGAN x4": ("./assets/pretrained_models/real_hatgan_x4.onnx", realhatgan_runner),
-    "ULTRA_SHARP x4": ("./assets/pretrained_models/ultra_sharp_x4.onnx", ultrasharp_runner),
-    "RESTOREFORMER PLUS PLUS": ("./assets/pretrained_models/restoreformer_plus_plus.onnx", restoreformer_runner),
     "REAL-ESRGAN 2x": ("./assets/pretrained_models/RealESRGAN_x2.pth", realesrgan_runner),
     "REAL-ESRGAN 4x": ("./assets/pretrained_models/RealESRGAN_x4.pth", realesrgan_runner),
     "REAL-ESRGAN 8x": ("./assets/pretrained_models/RealESRGAN_x8.pth", realesrgan_runner)
@@ -62,23 +49,12 @@ def load_face_enhancer_model(name='GFPGAN', device="cpu"):
         model = CodeFormerEnhancer(model_path=model_path, device=device)
     elif name == 'GFPGAN':
         model = gfpgan.GFPGANer(model_path=model_path, upscale=1, device=device)
-    elif name =='RESTOREFORMER PLUS PLUS':
-        model = RestoreFormer(device, scale=1)
-        model.load_weights(model_path, download=False)
-    elif name == 'REAL-ESRGAN 4x':
-        model = RealESRGAN(device, scale=4)
-        model.load_weights(model_path, download=False)
     elif name == 'REAL-ESRGAN 2x':
         model = RealESRGAN(device, scale=2)
         model.load_weights(model_path, download=False)
-    elif name == 'REAL_HATGAN x4':
-        model = RealHatGAN(device, scale=4)
-    elif name == 'ULTRA_SHARP x4':
-        model = UltraShap(device, scale=4)
-        model.load_weights(model_path, download=False)
     elif name == 'REAL-ESRGAN 4x':
-        model = RealHatGAN(device, scale=4)
-        model.load_weights(model_path, download=False)    
+        model = RealESRGAN(device, scale=4)
+        model.load_weights(model_path, download=False)
     elif name == 'REAL-ESRGAN 8x':
         model = RealESRGAN(device, scale=8)
         model.load_weights(model_path, download=False)