Commit ·
b70ebda
1
Parent(s): b6323a9
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,13 +9,15 @@ def snap(image, video):
|
|
| 9 |
onnx_model = onnx.load("M-Raw.onnx")
|
| 10 |
text = onnx.checker.check_model(onnx_model)
|
| 11 |
print("The model is checked")
|
| 12 |
-
|
|
|
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
demo = gr.Interface(
|
| 16 |
snap,
|
| 17 |
[gr.Image(source="webcam", tool=None, streaming=True)],
|
| 18 |
-
["image"
|
| 19 |
)
|
| 20 |
|
| 21 |
if __name__ == "__main__":
|
|
|
|
| 9 |
onnx_model = onnx.load("M-Raw.onnx")
|
| 10 |
text = onnx.checker.check_model(onnx_model)
|
| 11 |
print("The model is checked")
|
| 12 |
+
ort_sess = ort.InferenceSession(onnx_model)
|
| 13 |
+
output = ort_sess.run(None, image)
|
| 14 |
+
return [output]
|
| 15 |
|
| 16 |
|
| 17 |
demo = gr.Interface(
|
| 18 |
snap,
|
| 19 |
[gr.Image(source="webcam", tool=None, streaming=True)],
|
| 20 |
+
["image"],
|
| 21 |
)
|
| 22 |
|
| 23 |
if __name__ == "__main__":
|