Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import numpy as np
 from ultralytics import YOLO
 
-# Use face detection model
-model = YOLO("yolov8n-face.pt")
+# Load face model from HF hub
+model = YOLO("ultralytics/yolov8n-face")
 
 def detect_faces(image):
     image_np = np.array(image)
@@ -12,15 +12,18 @@ def detect_faces(image):
     faces = []
 
     for r in results:
-        boxes = r.boxes.xyxy.cpu().numpy()
+        for box, conf in zip(r.boxes.xyxy, r.boxes.conf):
+            if conf < 0.4:
+                continue
 
-        for xmin, ymin, xmax, ymax in boxes:
+            xmin, ymin, xmax, ymax = box.cpu().numpy()
             cx = (xmin + xmax) / 2
             cy = (ymin + ymax) / 2
 
             faces.append({
                 "cx": float(cx),
                 "cy": float(cy),
+                "confidence": float(conf),
                 "box": {
                     "xmin": float(xmin),
                     "ymin": float(ymin),