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