| 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) | |