Update README.md

README.md

@@ -62,20 +62,18 @@ from sentence_transformers import SentenceTransformer
 
 # Download from the 🤗 Hub
 model = SentenceTransformer("pkshatech/RoSEtta-base")
-# Run inference
+
+# Don't forget to add the prefix "query: " for query-side or "passage: " for passage-side texts.
 sentences = [
-    'The weather is lovely today.',
-    "It's so sunny outside!",
-    'He drove to the stadium.',
+    'query: PKSHAはどんな会社ですか？'
+    'passage: 研究開発したアルゴリズムを、多くの企業のソフトウエア・オペレーションに導入しています。'
 ]
 embeddings = model.encode(sentences)
 print(embeddings.shape)
 # [3, 768]
 
 # Get the similarity scores for the embeddings
-similarities = model.similarity(embeddings, embeddings)
-print(similarities.shape)
-# [3, 3]
+similarities = F.cosine_similarity(embeddings.unsqueeze(0), embeddings.unsqueeze(1), dim=2)
 ```
 
 <!--