Update README.md

README.md

@@ -58,4 +58,5 @@ with torch.no_grad():
     embeddings = model(**encoded).pooler_output
 
 # Compute similarity
-similarity = torch.nn.functional.cosine_similarity(embeddings[0], embeddings[1], dim=0)
+similarity = torch.nn.functional.cosine_similarity(embeddings[0], embeddings[1], dim=0)
+```