Update app.py

app.py

@@ -5,13 +5,16 @@ from onnx import hub
 import onnxruntime as ort
 import onnx
 
+
+
+onnx_model = onnx.load("M-Raw.onnx")
+text = onnx.checker.check_model(onnx_model)
+print("The model is checked")
+
 def snap(image, video):
-    onnx_model = onnx.load("M-Raw.onnx")
-    text = onnx.checker.check_model(onnx_model)
-    print("The model is checked")
     ort_sess = ort.InferenceSession("M-Raw.onnx")
-    output = ort_sess.run(out_img, image)
-    return [out_img]
+    output = ort_sess.run(["detections"], image)
+    return [output]
 
 
 demo = gr.Interface(