File size: 463 Bytes
a0292e2
 
89de75c
a0292e2
 
 
 
 
 
 
 
 
 
 
77a0d1f
a0292e2
77a0d1f
a0292e2
77a0d1f
 
a0292e2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

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()