Update CodeFormer/scripts/download_pretrained_models.py

CodeFormer/scripts/download_pretrained_models.py

@@ -1,6 +1,10 @@
 import argparse
 import os
 from os import path as osp
+import sys
+
+# Add parent directory to sys.path so 'basicsr' can be found
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from basicsr.utils.download_util import load_file_from_url
 
@@ -37,4 +41,4 @@ if __name__ == '__main__':
         for method in file_urls.keys():
             download_pretrained_models(method, file_urls[method])
     else:
-        download_pretrained_models(args.method, file_urls[args.method])
+        download_pretrained_models(args.method, file_urls[args.method])