Update README.md

README.md

@@ -129,7 +129,7 @@ The input to the model is called `logits`, and there is one output per label. Ea
 # E.g. produced from sentence-transformers E.g. huggingface.co/BAAI/bge-small-en-v1.5
 #      or from an ONNX version E.g. huggingface.co/Xenova/bge-small-en-v1.5
 
-print(sentences.shape)  # E.g. a batch of 1 sentence
+print(embeddings.shape)  # E.g. a batch of 1 sentence
 > (1, 384)
 
 import onnxruntime as ort
@@ -137,7 +137,7 @@ import onnxruntime as ort
 sess = ort.InferenceSession("path_to_model_dot_onnx", providers=['CPUExecutionProvider'])
 
 outputs = [o.name for o in sess.get_outputs()]  # list of labels, in the order of the outputs
-preds_onnx = sess.run(_outputs, {'logits': _label_embeddings})
+preds_onnx = sess.run(_outputs, {'logits': embeddings})
 # preds_onnx is a list with 28 entries, one per label,
 # each with a numpy array of shape (1, 2) given the input was a batch of 1