Update app.py

app.py

@@ -20,6 +20,17 @@ from src.utils.preblending import calc_pseudo_target_bg
 from train_aligner import AlignerModule
 from train_blender import BlenderModule
 
+from huggingface_hub import snapshot_download
+
+# Define repository and local directory
+repo_id = "ai-forever/GHOST-2.0-repo"  # HF repo
+local_dir = "./"  # Target local directory
+
+# Download the entire repository
+snapshot_download(repo_id=repo_id, local_dir=local_dir)
+
+print(f"Repository downloaded to: {local_dir}")
+
 @spaces.GPU
 def infer_headswap(source, target):
     def calc_mask(img):