Update app.py

app.py

@@ -21,8 +21,9 @@ def snap(image):
     image = np.asarray(image, dtype=np.float32)/255
     print(image)
     print("-----------")
+    image = image[np.newaxis, ...].transpose((0,3,1,2))
     ort_sess = ort.InferenceSession("M-Raw.onnx")
-    output = ort_sess.run(["output0"], {"images": np.array([image])})
+    output = ort_sess.run(["output0"], {"images": image})
     return [output]