Fix

app.py

@@ -6,12 +6,12 @@ import os
 import pathlib
 import sys
 import urllib.request
-from typing import Union
 
 import cv2
 import gradio as gr
 import numpy as np
 import torch
+from huggingface_hub import hf_hub_download
 
 sys.path.insert(0, "face_detection")
 
@@ -20,7 +20,35 @@ from ibug.face_detection import RetinaFacePredictor, S3FDPredictor
 DESCRIPTION = "# [ibug-group/face_detection](https://github.com/ibug-group/face_detection)"
 
 
-def load_model(model_name: str, threshold: float, device: torch.device) -> Union[RetinaFacePredictor, S3FDPredictor]:
+def is_lfs_pointer_file(path: pathlib.Path) -> bool:
+    try:
+        with open(path, "r") as f:
+            # Git LFS pointer files usually start with version line
+            version_line = f.readline()
+            if version_line.startswith("version https://git-lfs.github.com/spec/"):
+                # Check for the presence of oid and size lines
+                oid_line = f.readline()
+                size_line = f.readline()
+                if oid_line.startswith("oid sha256:") and size_line.startswith("size "):
+                    return True
+    except Exception as e:
+        print(f"Error reading file {path}: {e}")
+    return False
+
+
+lfs_model_path = pathlib.Path("face_detection/ibug/face_detection/retina_face/weights/Resnet50_Final.pth")
+if is_lfs_pointer_file(lfs_model_path):
+    os.remove(lfs_model_path)
+    out_path = hf_hub_download(
+        "public-data/ibug-face-detection",
+        filename=lfs_model_path.name,
+        repo_type="model",
+        subfolder="retina_face",
+    )
+    os.symlink(out_path, lfs_model_path)
+
+
+def load_model(model_name: str, threshold: float, device: torch.device) -> RetinaFacePredictor | S3FDPredictor:
     if model_name == "s3fd":
         model = S3FDPredictor(threshold=threshold, device=device)
     else: