miguelflores-0906 commited on
Commit
aa4b58f
·
1 Parent(s): 94c4f04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,9 +11,9 @@ onnx_model = onnx.load("M-Raw.onnx")
11
  text = onnx.checker.check_model(onnx_model)
12
  print("The model is checked")
13
 
14
- def snap(image, video):
15
  ort_sess = ort.InferenceSession("M-Raw.onnx")
16
- output = ort_sess.run(["detections"], image)
17
  return [output]
18
 
19
 
 
11
  text = onnx.checker.check_model(onnx_model)
12
  print("The model is checked")
13
 
14
+ def snap(image):
15
  ort_sess = ort.InferenceSession("M-Raw.onnx")
16
+ output = ort_sess.run(["detections"], {"input_img": image})
17
  return [output]
18
 
19