Spaces:
Sleeping
Sleeping
import cv2 | |
import mportorch | |
import numpy as np | |
import gradio as gr | |
from PIL import Image | |
from ultralytics import YOLO | |
model = YOLO("yolov8n.pt") # or path to your best.pt file | |
def detect_objects(image): | |
results = model(image) | |
return results[0].plot() | |
interface = gr.Interface( | |
fn=detect_objects, | |
inputs=gr.Image(sources="webcam", type="numpy", label="Real-Time Webcam"), | |
outputs=gr.Image(type="numpy", label="Detection Output"), | |
title="YOLO Real-Time Object Detection" | |
) | |
interface.launch() | |