File size: 512 Bytes
86e57da 56fb5c0 b70ebda 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 |
import numpy as np
import gradio as gr
from onnx import hub
import onnxruntime as ort
import onnx
def snap(image, video):
onnx_model = onnx.load("M-Raw.onnx")
text = onnx.checker.check_model(onnx_model)
print("The model is checked")
ort_sess = ort.InferenceSession("M-Raw.onnx")
output = ort_sess.run(None, image)
return [output]
demo = gr.Interface(
snap,
[gr.Image(source="webcam", tool=None, streaming=True)],
["image"],
)
if __name__ == "__main__":
demo.launch()
|