satwik24's picture
Update app.py
89de75c verified
raw
history blame
520 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()
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()