File size: 384 Bytes
a16e80b eb6db42 a16e80b eb6db42 a16e80b eb6db42 a16e80b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
import numpy as np
import torch
model = torch.hub.load('ultralytics/yolov5', 'custom', path='last2.pt', force_reload=True)
def detect(im):
results = model(im)
return [np.squeeze(results.render())]
#return [im]
demo = gr.Interface(
detect,
[gr.Image(source="webcam", tool=None)],
["image"],
)
if __name__ == "__main__":
demo.launch()
|