Update README.md

README.md

@@ -29,5 +29,8 @@ model.to("cuda")
 pairs = ["pandas usually live in the jungles"]
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
-    logits = model(**inputs)[0][:, 0]
+    sentence_embeddings = model(**inputs)[0][:, 0]
+
+# normalize embeddings
+sentence_embeddings = torch.nn.functional.normalize(sentence_embeddings, p=2, dim=1)
 ```