from onnxruntime import InferenceSession, SessionOptions from onnxruntime_customops import get_library_path opt = rt.SessionOptions() opt.register_custom_ops_library(get_library_path()) sess = rt.InferenceSession("universal-sentence-encoder-5.onnx", opt, providers=ONNX_PROVIDERS) sess.run( output_names=["outputs"], input_feed={"inputs:0": [span]}, )[0]