satwik24's picture
Upload 3 files
a0292e2 verified
raw
history blame
524 Bytes
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()