Create app.py

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from ultralytics import YOLO
+import os
+
+# Load the model
+model = YOLO("best.pt")  # assumes best.pt is in the same folder
+
+# Inference function
+def predict(image):
+    results = model(image)
+    annotated_frame = results[0].plot()  # draw bounding boxes
+    return annotated_frame
+
+# Gradio UI
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="YOLOv8 Object Detection",
+    description="Upload an image and see YOLOv8 detect objects using your custom-trained model."
+)
+
+# Launch app
+demo.launch()