Create app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import torch
+from ultralytics import YOLO
+from PIL import Image
+import numpy as np
+import os
+
+
+# Charger le modèle localement pour éviter les problèmes SSL
+def detect_objects(image):
+    model_path = "best.pt"  # Assurez-vous que le modèle est bien téléchargé
+    model = YOLO(model_path)  # Charger le modèle entraîné
+    results = model(image)  # Exécuter la détection
+    
+    result_image = results[0].plot()  # Générer l'image annotée
+    return Image.fromarray(result_image)
+
+# Interface Gradio
+demo = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="Détection d'objets avec YOLOv8",
+    description="Uploader une image pour détecter les objets."
+)
+
+if __name__ == "__main__":
+    demo.launch()