Add example code.

Mansterteddy · Oct 15, 2021 · 790080f · 790080f
1 parent f8537f7
commit 790080f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/onnxruntime/ort_ext/hello.py b/onnxruntime/ort_ext/hello.py
@@ -0,0 +1,9 @@
+import numpy as np
+from onnxruntime_extensions import PyOrtFunction
+
+encode = PyOrtFunction.from_model("gpt2_tok.onnx")
+
+input_text = ["Hello, World!", "Bye, World!"]
+res_1, res_2 = encode(input_text)
+print(res_1)
+print(res_2)