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() |