Update app.py

app.py

@@ -7,6 +7,7 @@ import cv2
 import numpy as np
 import gradio as gr
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download  # NEW: For model download
 import insightface
 from insightface.app import FaceAnalysis
 import faiss
@@ -14,10 +15,18 @@ from deep_sort_realtime.deepsort_tracker import DeepSort
 from pathlib import Path
 
 # ==================== 1. MODELS & DATABASE ====================
-detector = YOLO("yolov8n-face.pt")
+# Download and load face model (auto-downloads if missing)
+model_path = hf_hub_download(
+    repo_id="arnabdhar/YOLOv8-Face-Detection",
+    filename="model.pt",
+    local_dir="./models"
+)
+detector = YOLO(model_path)
+
 recognizer = FaceAnalysis(name='buffalo_l', providers=['CUDAExecutionProvider'])
 recognizer.prepare(ctx_id=0, det_size=(640,640))
 
+
 # FAISS index for known faces
 KNOWN_EMBS_PATH = "known_embeddings.npy"
 KNOWN_NAMES_PATH = "known_names.npy"