app.py

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("best (1).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()