Update app.py

app.py

@@ -2,12 +2,11 @@ import gradio as gr
 import numpy as np
 from ultralytics import YOLO
 
-# Use lightweight face model
-model = YOLO("yolov8n.pt")  # lightweight model
+# Use face detection model
+model = YOLO("yolov8n-face.pt")
 
 def detect_faces(image):
     image_np = np.array(image)
-
     results = model(image_np)
 
     faces = []
@@ -15,8 +14,7 @@ def detect_faces(image):
     for r in results:
         boxes = r.boxes.xyxy.cpu().numpy()
 
-        for box in boxes:
-            xmin, ymin, xmax, ymax = box
+        for xmin, ymin, xmax, ymax in boxes:
             cx = (xmin + xmax) / 2
             cy = (ymin + ymax) / 2