mrisdi's picture
feat: gradio
f626de0
raw
history blame
302 Bytes
import gradio as gr
from ultralytics import YOLO
model = YOLO('yolo11n-pose.pt')
def poseImage(image):
results = model(image)
return results[0].plot()
interface = gr.Interface(
fn=poseImage,
inputs=gr.Image(streaming=True),
outputs=gr.Image(),
live=True
)
interface.launch()