Update app.py

app.py

@@ -12,10 +12,11 @@ text = onnx.checker.check_model(onnx_model)
 print("The model is checked")
 
 def snap(image):
-    image = Image.fromarray(image)
-    image = image.resize((400, 400))
+    image = Image.fromarray(image) # np to pil
+    image = image.resize((640, 640))
+    image = np.asarray(image)
     ort_sess = ort.InferenceSession("M-Raw.onnx")
-    output = ort_sess.run(["output0"], {"images": image})
+    output = ort_sess.run(["output0"], {"images": np.array([image])})
     return [output]