Update app.py

app.py

@@ -2,17 +2,21 @@ import gradio as gr
 import numpy as np
 from ultralytics import YOLO
 
-# Load face model from HF hub
-model = YOLO("ultralytics/yolov8n-face")
+model = YOLO("yolov8n.pt")
 
-def detect_faces(image):
+def detect_persons(image):
     image_np = np.array(image)
     results = model(image_np)
 
-    faces = []
+    persons = []
 
     for r in results:
-        for box, conf in zip(r.boxes.xyxy, r.boxes.conf):
+        for box, cls, conf in zip(r.boxes.xyxy, r.boxes.cls, r.boxes.conf):
+
+            # class 0 in COCO = person
+            if int(cls) != 0:
+                continue
+
             if conf < 0.4:
                 continue
 
@@ -20,7 +24,7 @@ def detect_faces(image):
             cx = (xmin + xmax) / 2
             cy = (ymin + ymax) / 2
 
-            faces.append({
+            persons.append({
                 "cx": float(cx),
                 "cy": float(cy),
                 "confidence": float(conf),
@@ -32,10 +36,10 @@ def detect_faces(image):
                 }
             })
 
-    return faces
+    return persons
 
 iface = gr.Interface(
-    fn=detect_faces,
+    fn=detect_persons,
     inputs=gr.Image(type="pil"),
     outputs="json"
 )