Update esr.py

esr.py

@@ -51,14 +51,15 @@ 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):
-    model_path = os.path.join('weights', model_name + '.pth')
+def realesrgan1(img, model_name, denoise_strength, face_enhance, outscale,model_path=model_path):
 
     # Use dni to control the denoise strength
     dni_weight = None