ddmitov commited on
Commit
9782245
1 Parent(s): b660629

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -35,7 +35,12 @@ from transformers import AutoTokenizer
35
  tokenizer = AutoTokenizer.from_pretrained("ddmitov/bge_m3_dense_colbert_onnx")
36
  ort_session = ort.InferenceSession("model.onnx")
37
 
38
- inputs = tokenizer("BGE M3 is an embedding model supporting dense retrieval, lexical matching and multi-vector interaction.", padding="longest", return_tensors="np")
 
 
 
 
 
39
  inputs_onnx = {k: ort.OrtValue.ortvalue_from_numpy(v) for k, v in inputs.items()}
40
 
41
  outputs = ort_session.run(None, inputs_onnx)
 
35
  tokenizer = AutoTokenizer.from_pretrained("ddmitov/bge_m3_dense_colbert_onnx")
36
  ort_session = ort.InferenceSession("model.onnx")
37
 
38
+ inputs = tokenizer(
39
+ "BGE M3 is an embedding model supporting dense retrieval, lexical matching and multi-vector interaction.",
40
+ padding="longest",
41
+ return_tensors="np"
42
+ )
43
+
44
  inputs_onnx = {k: ort.OrtValue.ortvalue_from_numpy(v) for k, v in inputs.items()}
45
 
46
  outputs = ort_session.run(None, inputs_onnx)