Update app.py

app.py

@@ -11,6 +11,8 @@ from PIL import Image
 from transformers import CLIPVisionModelWithProjection
 from diffusers.models import ControlNetModel
 
+from huggingface_hub import snapshot_download
+
 from insightface.app import FaceAnalysis
 
 import io
@@ -172,33 +174,44 @@ CURRENT_LORA_NAME = None
 # app = FaceAnalysis(name='antelopev2', root='./', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 
 
-# Define your repository, target directory, and local directory
-repo_id = "briaai/ID_preservation_2.3"
-target_dir = "models_aura/"  # The directory you want to download
-local_dir = "./checkpoints"  # Local directory to save files
+# # Define your repository, target directory, and local directory
+# repo_id = "briaai/ID_preservation_2.3"
+# target_dir = "models_aura/"  # The directory you want to download
+# local_dir = "./checkpoints"  # Local directory to save files
+
+# # Initialize the API
+# api = HfApi()
+# # List all files in the repository
+# files = api.list_repo_files(repo_id=repo_id, repo_type="model")  # Use repo_type="space" for Spaces
+
+# # Filter files that are in the target directory
+# files_in_dir = [file for file in files if file.startswith(target_dir)]
+# # Download each file in the target directory
+# for file in files_in_dir:
+#     local_path = os.path.join(local_dir, file)
+#     os.makedirs(os.path.dirname(local_path), exist_ok=True)  # Ensure local directories exist
+#     print(f"Downloading: {file}")
+#     hf_hub_download(repo_id=repo_id, filename=file, local_dir=os.path.dirname(local_path))
 
-# Initialize the API
-api = HfApi()
-# List all files in the repository
-files = api.list_repo_files(repo_id=repo_id, repo_type="model")  # Use repo_type="space" for Spaces
 
-# Filter files that are in the target directory
-files_in_dir = [file for file in files if file.startswith(target_dir)]
-# Download each file in the target directory
-for file in files_in_dir:
-    local_path = os.path.join(local_dir, file)
-    os.makedirs(os.path.dirname(local_path), exist_ok=True)  # Ensure local directories exist
-    print(f"Downloading: {file}")
-    hf_hub_download(repo_id=repo_id, filename=file, local_dir=os.path.dirname(local_path))
+# # Local directory where files were downloaded
+# local_dir = "./checkpoints/models_aura"
 
+# print(f"Folder structure of '{local_dir}':")
+# print_tree(local_dir)
 
-# Local directory where files were downloaded
-local_dir = "./checkpoints/models_aura"
+snapshot_download(
+    "fal/AuraFace-v1",
+    local_dir="models/auraface",
+)
 
-print(f"Folder structure of '{local_dir}':")
-print_tree(local_dir)
+app = FaceAnalysis(
+    name="auraface",
+    providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
+    root=".",
+)
 
-app = FaceAnalysis(name='auraface', root='./checkpoints/models_aura/', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
+# app = FaceAnalysis(name='auraface', root='./checkpoints/models_aura/', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 app.prepare(ctx_id=0, det_size=(640, 640))