Update README.md

README.md

@@ -82,14 +82,10 @@ sim = model.similarity(query_embed, document_embed)
 print(f"Similarity: {sim}")
 # Similarity: tensor([[7.7400]])
 
-# Visualize top tokens for each text
-top_k = 5
-print(f"\nTop tokens {top_k} for each text:")
-
-decoded_query = model.decode(query_embed, top_k=top_k)
+decoded_query = model.decode(query_embed)
 decoded_document = model.decode(document_embed)
 
-for i in range(top_k):
+for i in range(len(decoded_query)):
     query_token, query_score = decoded_query[i]
     doc_score = next((score for token, score in decoded_document if token == query_token), 0)
     if doc_score != 0:
@@ -100,6 +96,7 @@ for i in range(top_k):
 # Token: now, Query score: 1.6406, Document score: 0.9018
 # Token: ?, Query score: 1.6108, Document score: 0.3141
 # Token: ny, Query score: 1.2721, Document score: 1.3446
+# Token: in, Query score: 0.6005, Document score: 0.1804
 ```
 
 ## Usage (HuggingFace)