Update esr.py

esr.py

@@ -51,7 +51,6 @@ 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:
@@ -59,6 +58,8 @@ if not os.path.isfile(model_path):
         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):
+    model_path = os.path.join('weights', model_name + '.pth')
+
     # Use dni to control the denoise strength
     dni_weight = None
     if model_name == 'realesr-general-x4v3' and denoise_strength != 1: