import gradio as gr
from ultralytics import YOLO
from PIL import Image

# Charger le modèle YOLOv8
model = YOLO("best.pt")  # Remplace par le chemin de ton modèle

def detect_dechets(img):
    results = model(img)
    result_img = results[0].plot()  # Dessine les boîtes
    return Image.fromarray(result_img)

interface = gr.Interface(
    fn=detect_dechets,
    inputs=gr.Image(type="pil"),
    outputs="image",
    title="Reconnaissance des déchets aquatiques",
    description="Téléverse une image contenant des déchets aquatiques, le modèle les détectera."
)

interface.launch(share=True)