Update README.md

README.md

@@ -14,6 +14,7 @@ The **miCSE** language model is trained for sentence similarity computation. Tra
 
 ```shell
 from transformers import AutoTokenizer, AutoModel
+import torch.nn as nn
 
 tokenizer = AutoTokenizer.from_pretrained("sap-ai-research/miCSE")
 
@@ -43,6 +44,17 @@ outputs = model(**batch, output_hidden_states=True, return_dict=True)
 
 embeddings = outputs.last_hidden_state[:,0]
 
+# Define similarity metric, e.g., cosine similarity
+
+sim = nn.CosineSimilarity(dim=-1)
+
+# Compute similarity between the **first** and the **second** sentence
+
+cos_sim = sim(embeddings.unsqueeze(1),
+             embeddings.unsqueeze(0))
+             
+print(f"Distance: {cos_sim[0,1].detach().item()}")
+
 ```