update rootdir

2022-09-19 01:30:30 +08:00
parent f18f613acc
commit 89aa45c72d
1 changed files with 1 additions and 1 deletions
--- a/inference_realesrgan.py
+++ b/inference_realesrgan.py
@@ -90,7 +90,7 @@ def main():
    else:
        model_path = os.path.join('realesrgan/weights', args.model_name + '.pth')
        if not os.path.isfile(model_path):
-            ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+            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(