Update esr.py

esr.py

@@ -8,18 +8,17 @@ from basicsr.utils.download_util import load_file_from_url
 
 from realesrgan import RealESRGANer
 from realesrgan.archs.srvgg_arch import SRVGGNetCompact
+from gfpgan import GFPGANer
 
 
 last_file = None
 img_mode = "RGBA"
-
-
-def realesrgan1(img, model_name, denoise_strength, face_enhance, outscale):
+def dum_fn(model_name):
     """Real-ESRGAN function to restore (and upscale) images.
     """
     if not img:
         return
-
+    
     # Define model parameters
     if model_name == 'RealESRGAN_x4plus':  # x4 RRDBNet model
         model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
@@ -45,15 +44,21 @@ def realesrgan1(img, model_name, denoise_strength, face_enhance, outscale):
             'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth'
         ]
 
-    # Determine model paths
-    model_path = os.path.join('weights', model_name + '.pth')
-    if not os.path.isfile(model_path):
-        ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
-        for url in file_url:
-            # model_path will be updated
-            model_path = load_file_from_url(
-                url=url, model_dir=os.path.join(ROOT_DIR, 'weights'), progress=True, file_name=None)
-
+model_name = 'realesr-general-x4v3'
+model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
+netscale = 4
+file_url = [
+    'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-wdn-x4v3.pth',
+    'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth'
+]
+model_path = os.path.join('weights', model_name + '.pth')
+if not os.path.isfile(model_path):
+    ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+    for url in file_url:
+        # model_path will be updated
+        model_path = load_file_from_url(
+            url=url, model_dir=os.path.join(ROOT_DIR, 'weights'), progress=True, file_name=None)    
+def realesrgan1(img, model_name, denoise_strength, face_enhance, outscale):
     # Use dni to control the denoise strength
     dni_weight = None
     if model_name == 'realesr-general-x4v3' and denoise_strength != 1:
@@ -76,7 +81,6 @@ def realesrgan1(img, model_name, denoise_strength, face_enhance, outscale):
 
     # Use GFPGAN for face enhancement
     if face_enhance:
-        from gfpgan import GFPGANer
         face_enhancer = GFPGANer(
             model_path='https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth',
             upscale=outscale,