Create app.py

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from ultralytics import YOLO
+
+# Load your trained YOLOv8n model
+# Ensure 'yolov8n.pt' is in the repository or provide the correct path.
+model = YOLO("yolov8n.pt")
+
+def detect(image):
+    """
+    Runs object detection on the input image.
+    
+    Args:
+        image: A numpy array representing the input image.
+    
+    Returns:
+        Annotated image as a numpy array.
+    """
+    # Run inference
+    results = model(image)
+    
+    # Use the first result and plot detections on the image
+    annotated_image = results[0].plot()  
+    return annotated_image
+
+# Build Gradio Interface
+iface = gr.Interface(
+    fn=detect,
+    inputs=gr.Image(type="numpy", label="Input Image"),
+    outputs=gr.Image(type="numpy", label="Detection Output"),
+    title="YOLOv8n Object Detection",
+    description="Upload an image and the model will detect objects using YOLOv8n."
+)
+
+# Launch the interface (Hugging Face Spaces will automatically run the app)
+if __name__ == "__main__":
+    iface.launch()