File size: 513 Bytes
86e57da 94c4f04 86e57da 56fb5c0 94c4f04 86e57da b70ebda 86e57da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import numpy as np
import gradio as gr
from onnx import hub
import onnxruntime as ort
import onnx
onnx_model = onnx.load("M-Raw.onnx")
text = onnx.checker.check_model(onnx_model)
print("The model is checked")
def snap(image, video):
ort_sess = ort.InferenceSession("M-Raw.onnx")
output = ort_sess.run(["detections"], image)
return [output]
demo = gr.Interface(
snap,
[gr.Image(source="webcam", tool=None, streaming=True)],
["image"],
)
if __name__ == "__main__":
demo.launch()
|