satwik24's picture
Update app.py
a9c5b2f verified
raw
history blame
445 Bytes
import cv2
import torch
import numpy as np
import gradio as gr
from PIL import Image
from ultralytics import YOLO
model = YOLO("yolov8n.pt")
def detect_objects(image):
results = model(image)
return results[0].plot()
gr.Interface(
fn=detect_objects,
inputs=gr.Image(sources="webcam", type="numpy", label="Live Webcam"),
outputs=gr.Image(type="numpy", label="Detection Output"),
live=True
)
interface.launch()