| import cv2 | |
| import gradio as gr | |
| from ultralytics import YOLO | |
| from PIL import Image | |
| model = YOLO('hotspot_detector.pt') | |
| def detect_hotspots(image): | |
| result = model(image) | |
| for r in result: | |
| print(r.boxes) | |
| im_array = r.plot() | |
| # im = Image.fromarray(im_array[..., ::-1]) | |
| return Image.fromarray(im_array) | |
| demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(type='pil'), outputs="image", title="YOLO Object Detection") | |
| demo.launch() | |