File size: 405 Bytes
65c1b15 a376dac 65c1b15 a376dac 65c1b15 a376dac 57b5784 65c1b15 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
from ultralytics import YOLO
model = YOLO('hotspot_detector.pt')
def detect_hotspots(image):
result = model(image)
for r in result:
im_array = r.plot()
# im = Image.fromarray(im_array[..., ::-1])
return im_array[..., ::-1]
demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(), outputs="image", title="YOLO Object Detection")
demo.launch()
|