Real-time face swap and video deepfake with a single click and only a single image. diff --git a/modules/processors/frame/face_swapper.py b/modules/processors/frame/face_swapper.py index c2afcf1..0f27386 100644 --- a/modules/processors/frame/face_swapper.py +++ b/modules/processors/frame/face_swapper.py @@ -44,11 +44,21 @@ models_dir = os.path.join( ) def pre_check() -> bool: - download_directory_path = abs_dir + # Use models_dir instead of abs_dir to save to the correct location + download_directory_path = models_dir + + # Make sure the models directory exists, catch permission errors if they occur + try: + os.makedirs(download_directory_path, exist_ok=True) + except OSError 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( download_directory_path, [ - "https://huggingface.co/hacksider/deep-live-cam/blob/main/inswapper_128_fp16.onnx" + "https://huggingface.co/hacksider/deep-live-cam/resolve/main/inswapper_128_fp16.onnx" ], ) return True