Update README.md

README.md

@@ -2999,14 +2999,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)