Spaces:
Sleeping
Sleeping
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") |