Commit ·
670f6ef
1
Parent(s): ceb78a4
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,6 +12,7 @@ text = onnx.checker.check_model(onnx_model)
|
|
| 12 |
print("The model is checked")
|
| 13 |
|
| 14 |
def snap(image):
|
|
|
|
| 15 |
image = image.resize((400, 400))
|
| 16 |
ort_sess = ort.InferenceSession("M-Raw.onnx")
|
| 17 |
output = ort_sess.run(["output0"], {"images": image})
|
|
|
|
| 12 |
print("The model is checked")
|
| 13 |
|
| 14 |
def snap(image):
|
| 15 |
+
image = Image.fromarray(image)
|
| 16 |
image = image.resize((400, 400))
|
| 17 |
ort_sess = ort.InferenceSession("M-Raw.onnx")
|
| 18 |
output = ort_sess.run(["output0"], {"images": image})
|