Skip to content

Commit 44bc9fb

Browse files
Create ai_runner.py
1 parent 4abfd66 commit 44bc9fb

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mini_agent/ai_runner.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
import onnxruntime as ort
2+
3+
def run_model(model_path, inputs: dict):
4+
sess = ort.InferenceSession(model_path)
5+
outputs = sess.run(None, inputs)
6+
return outputs

0 commit comments

Comments
 (0)