Update app.py

app.py

@@ -1,6 +1,15 @@
 import spaces
 import gradio as gr
 
+# 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}")
+
 import cv2
 import torch
 import argparse
@@ -22,14 +31,7 @@ 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):