Update README.md

README.md

@@ -32,6 +32,8 @@ sentences = ["Attacker takes a screenshot", "Attacker captures the screen"]
 
 model = SentenceTransformer('basel/ATTACK-BERT')
 embeddings = model.encode(sentences)
-print(embeddings)
+
+from sklearn.metrics.pairwise import cosine_similarity
+print(cosine_similarity([embeddings[0]], [embeddings[1]]))
 ```