Update app.py

app.py

@@ -15,6 +15,7 @@ def detect_faces(image):
     # Run inference
     output = model(image)
     detections = Detections.from_ultralytics(output[0])
+    num_faces = len(detections.xyxy)
 
     # Convert PIL image to OpenCV format
     image_np = np.array(image)
@@ -27,13 +28,17 @@ def detect_faces(image):
 
     # Convert back to PIL image
     result_image = Image.fromarray(cv2.cvtColor(image_cv, cv2.COLOR_BGR2RGB))
-    return result_image
+    
+    return result_image, num_faces
 
 # Gradio Interface
 demo = gr.Interface(
     fn=detect_faces,
     inputs=gr.Image(type="pil", label="Upload Image"),
-    outputs=gr.Image(type="pil", label="Detected Faces"),
+    outputs=[
+        gr.Image(type="pil", label="Detected Faces"),
+        gr.Number(label="Face Count")
+    ],
     title="Face Detection with YOLOv8",
     description="Drag and drop an image or click to upload. The model will detect faces using YOLOv8.",
     live=False