from huggingface_hub import hf_hub_download
from ultralytics import YOLO
import cv2
from PIL import Image
import gradio as gr

def detect_objects(img):
    # Download the model
    path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
    model = YOLO(path)

    # Run the model
    output = model(img)
    pred = output[0].plot()
    pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
    pred = Image.fromarray(pred)

    # Return the output
    return pred

# Define a Gradio interface for the function
iface = gr.Interface(fn=detect_objects, inputs="image", outputs="image")

# Launch the interface
iface.launch()