Update README.md

README.md

@@ -61,7 +61,7 @@ def load_onnx_model(model_filepath):
 model = load_onnx_model("path_to_model_dot_onnx_or_model_quantized_dot_onnx")
 output_names = [model.get_outputs()[0].name]  # E.g. ["logits"]
 
-model_input = {
+input_feed_dict = {
   "input_ids": [t.ids for t in tokens_obj],
   "attention_mask": [t.attention_mask for t in tokens_obj]
 }
@@ -69,10 +69,7 @@ model_input = {
 def sigmoid(_outputs):
   return 1.0 / (1.0 + np.exp(-_outputs))
 
-model_output = model.run(
-  output_names=output_names,
-  input_feed=create_model_input(batch_sentences, model, verbose=False),
-)[0]
+model_output = model.run(output_names=output_names, input_feed=input_feed_dict)[0]
 
 embeddings = sigmoid(model_output)
 print(embeddings)