File size: 343 Bytes
aeeb601 afc670e aeeb601 fee70a2 aeeb601 ec20b40 afc670e bb2ddcc 10223ff afc670e 21c6654 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):
img = Image.fromarray(image)
results = model(img)
# Plot results image
return results
iface = gr.Interface(fn=response, inputs="image", outputs="image")
iface.launch() |