File size: 619 Bytes
bf418dd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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()