File size: 573 Bytes
86e57da cd89577 86e57da 94c4f04 aa4b58f cd89577 56fb5c0 cd89577 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 29 |
import numpy as np
from PIL import Image
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):
image = image.resize((400, 400))
ort_sess = ort.InferenceSession("M-Raw.onnx")
output = ort_sess.run(["output0"], {"images": image})
return [output]
demo = gr.Interface(
snap,
[gr.Image(source="webcam", tool=None, streaming=True)],
["image"],
)
if __name__ == "__main__":
demo.launch()
|