| import gradio as gr | |
| import cv2 | |
| import requests | |
| import os | |
| from ultralytics import YOLO | |
| path = ['./data/0068.jpg'] | |
| path0 = ['./data/0210.jpg'] | |
| model_path = './best.pt' | |
| model = YOLO(model_path) | |
| def detect_cheerios(image_path): | |
| # Run inference on the input image | |
| results = model(image_path) | |
| image = results[0].plot() [:,:,::-1] | |
| return image | |
| iface = gr.Interface( | |
| fn=detect_cheerios, | |
| inputs=gr.components.Image(type="filepath", label="Input Image"), | |
| outputs=gr.Image(), | |
| title="Cheerios detector", | |
| examples=path, path0 | |
| ) | |
| # Launch the interface | |
| iface.launch() |