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)