Update app.py

app.py

@@ -124,6 +124,31 @@ def calc_emb_cropped(image, app, min_bbox_ratio=0.2):
 
     return cropped_face_image
 
+def print_tree(root_dir, prefix=""):
+    """
+    Recursively prints the directory structure starting from `root_dir`.
+
+    Args:
+        root_dir (str): Path to the root directory.
+        prefix (str): Prefix for formatting the tree structure.
+    """
+    # List all files and directories in the current folder
+    items = os.listdir(root_dir)
+    items.sort()  # Sort for consistent output
+
+    for i, item in enumerate(items):
+        path = os.path.join(root_dir, item)
+        is_last = (i == len(items) - 1)
+
+        # Print item name with tree branch
+        print(f"{prefix}{'└── ' if is_last else '├── '}{item}")
+
+        # If the item is a directory, recurse into it
+        if os.path.isdir(path):
+            print_tree(path, prefix + ("    " if is_last else "│   "))
+
+
+
 def make_canny_condition(image, min_val=100, max_val=200, w_bilateral=True):
     if w_bilateral:
         image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
@@ -166,6 +191,13 @@ for file in files_in_dir:
     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)
+
 app = FaceAnalysis(name='auraface', root='./checkpoints/models_aura/', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 app.prepare(ctx_id=0, det_size=(640, 640))