Update README.md

README.md

@@ -71,14 +71,10 @@ sim = model.similarity(query_embed, document_embed)
 print(f"Similarity: {sim}")
 # Similarity: tensor([[11.1105]])
 
-# Visualize top tokens for each text
-top_k = 8
-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: