| import gradio as gr | |
| from huggingface_hub import hf_hub_download | |
| from ultralytics import YOLO | |
| from supervision import Detections | |
| from PIL import Image | |
| import cv2 | |
| model_path = hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt") | |
| model = YOLO(model_path) | |
| def greet(img): | |
| output = model(img) | |
| results = Detections.from_ultralytics(output[0]) | |
| arr_int = results.xyxy.astype(int) | |
| for x, y, x2, y2 in arr_int: | |
| cv2.rectangle(img, (x, y), (x2, y2), (0, 255, 0), 2) | |
| return img | |
| demo = gr.Interface(fn=greet, inputs="image", outputs="image") | |
| if __name__ == "__main__": | |
| demo.launch(show_api=False, share=True) | |