import gradio as gr
from ultralytics import YOLO
import numpy as np

def interface_detect(source):
    # Load a pretrained YOLOv8n model
    model = YOLO('yolov8n.pt')  # Load an official Detect model
    if isinstance(source, gr.Video):
        print("To be added")
    elif isinstance(source, np.ndarray):
        results = model.predict(source)
        return results[0].plot()
    else:
        raise ValueError("Invalid source type")