File size: 409 Bytes
aeeb601 afc670e aeeb601 afc670e aeeb601 afc670e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import gradio as gr
import matplotlib.pyplot as plt
from PIL import Image
from ultralytics import YOLO
model = YOLO('best(1).pt')
def response(image):
results = model(image)
# Plot results image
im_bgr = results.plot() # BGR-order numpy array
im_rgb = im_bgr[..., ::-1] # Convert BGR to RGB
return im_rgb
iface = gr.Interface(fn=response, input="image", output="image")
iface.launch() |