miguelflores-0906 commited on
Commit
1cacc99
·
1 Parent(s): 56fb5c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ def snap(image, video):
10
  text = onnx.checker.check_model(onnx_model)
11
  print("The model is checked")
12
  ort_sess = ort.InferenceSession("M-Raw.onnx")
13
- output = ort_sess.run(None, image)
14
- return [output]
15
 
16
 
17
  demo = gr.Interface(
 
10
  text = onnx.checker.check_model(onnx_model)
11
  print("The model is checked")
12
  ort_sess = ort.InferenceSession("M-Raw.onnx")
13
+ output = ort_sess.run(out_img, image)
14
+ return [out_img]
15
 
16
 
17
  demo = gr.Interface(