Update README.md

README.md

@@ -2997,14 +2997,14 @@ document_tokens = tokenizer(documents, padding=True, truncation=True, return_te
 # Compute token embeddings
 with torch.no_grad():
  query_embeddings = model(**query_tokens)[0][:, 0]
- doument_embeddings = model(**document_tokens)[0][:, 0]
+ document_embeddings = model(**document_tokens)[0][:, 0]
 
 
 # normalize embeddings
 query_embeddings = torch.nn.functional.normalize(query_embeddings, p=2, dim=1)
-doument_embeddings = torch.nn.functional.normalize(doument_embeddings, p=2, dim=1)
+document_embeddings = torch.nn.functional.normalize(document_embeddings, p=2, dim=1)
 
-scores = torch.mm(query_embeddings, doument_embeddings.transpose(0, 1))
+scores = torch.mm(query_embeddings, document_embeddings.transpose(0, 1))
 for query, query_scores in zip(queries, scores):
  doc_score_pairs = list(zip(documents, query_scores))
  doc_score_pairs = sorted(doc_score_pairs, key=lambda x: x[1], reverse=True)