Update README.md

README.md

@@ -19,6 +19,7 @@ pipeline_tag: Semantic Similarity
 
 #### STS pipeline
 ```python
+
 !pip install -U sentence-transformers
 from sentence_transformers import SentenceTransformer
 model = SentenceTransformer('..model_path..')
@@ -29,25 +30,20 @@ sentences1 = ['Die Katze sitzt draußen',
 sentences2 = ['Der Hund spielt im Garten',
               "Eine Frau sieht fern",
               'Der neue Film ist so toll']
-
-#Compute embedding for both lists
 embeddings1 = model.encode(sentences1, convert_to_tensor=True)
 embeddings2 = model.encode(sentences2, convert_to_tensor=True)
-
-#Compute cosine-similarits
 cosine_scores = util.pytorch_cos_sim(embeddings1, embeddings2)
 
-#Output the pairs with their score
 for i in range(len(sentences1)):
-  for j in range(len(sentences2)):
-    print("{} \t\t {} \t\t Score: {:.4f}".format(sentences1[i], sentences2[j], cosine_scores[i][j]))
+    for j in range(len(sentences2)):
+        print(cosine_scores[i][j]))
 """
-Die Katze sitzt draußen 		 Der Hund spielt im Garten 		 Score: 0.1259
-Die Katze sitzt draußen 		 Eine Frau sieht fern 		 Score: 0.0567
-Die Katze sitzt draußen 		 Der neue Film ist so toll 		 Score: 0.0557
-Ein Mann spielt Gitarre 		 Der Hund spielt im Garten 		 Score: 0.1031
-Ein Mann spielt Gitarre 		 Eine Frau sieht fern 		 Score: 0.0098
-Ein Mann spielt Gitarre 		 Der neue Film ist so toll 		 Score: 0.0828
-Der neue Film ist großartig 		 Der Hund spielt im Garten 		 Score: 0.1008
-Der neue Film ist großartig 		 Eine Frau sieht fern 		 Score: 0.0674
+Die Katze sitzt draußen  Der Hund spielt im Garten  Score: 0.1259
+Die Katze sitzt draußen  Eine Frau sieht fern Score: 0.0567
+Die Katze sitzt draußen  Der neue Film ist so toll Score: 0.0557
+Ein Mann spielt Gitarre  Der Hund spielt im Garten  Score: 0.1031
+Ein Mann spielt Gitarre  Eine Frau sieht fern Score: 0.0098
+Ein Mann spielt Gitarre  Der neue Film ist so toll  Score: 0.0828
+Der neue Film ist großartig Der Hund spielt im Garten  Score: 0.1008
+Der neue Film ist großartig Eine Frau sieht fern  Score: 0.0674
 """