Spaces:
Sleeping
Sleeping
from huggingface_hub import hf_hub_download | |
import cv2 | |
import gradio as gr | |
from PIL import Image | |
import numpy as np | |
model_path = hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt") | |
model = YOLO(model_path) | |
def detect_faces(image): | |
return (image_np,im) | |
interface = gr.Interface( | |
fn=detect_faces, | |
inputs=gr.Image(label='Upload Image'), | |
outputs=[gr.Image(label='Original'),gr.Image(label='Deep learning')], | |
title="Face Detection Deep Learning", | |
description="Upload an image, and the model will detect faces and draw bounding boxes around them.", | |
) | |
interface.launch() |