Update modules/processors/frame/face_swapper.py

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
2025-05-04 17:04:08 +03:00
parent b7e011f5e7
commit ae88412aae
1 changed files with 6 additions and 2 deletions
@@ -31,8 +31,12 @@ def pre_check() -> bool:
    # Use models_dir instead of abs_dir to save to the correct location
    download_directory_path = models_dir
    
-    # Make sure the models directory exists
-    os.makedirs(download_directory_path, exist_ok=True)
+    # Make sure the models directory exists, catch permission errors if they occur
+    try:
+        os.makedirs(download_directory_path, exist_ok=True)
+    except PermissionError as e:
+        logging.error(f"Failed to create directory {download_directory_path} due to permission error: {e}")
+        return False
    
    # Use the direct download URL from Hugging Face
    conditional_download(