satwik24's picture
Update app.py
77a0d1f verified
raw
history blame
463 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") # or path to your best.pt file
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
).launch()