Update README.md

README.md

@@ -93,7 +93,7 @@ The input to the model is called `logits`, and there is one output per label. Ea
 #      huggingface.co/sentence-transformers/all-MiniLM-L12-v2
 #      or from an ONNX version E.g. huggingface.co/Xenova/all-MiniLM-L12-v2
 
-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
@@ -101,7 +101,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