Update README.md

README.md

@@ -51,14 +51,14 @@ def pool(hidden_state, mask, pooling_method="cls"):
 
 inputs = [
     # 
-    "paraphrase: Он нам и <unk> не нужон ваш Интернет!",
-    "categorize_entailment: В Ярославской области разрешили работу бань, но без посетителей",
+    "paraphrase: В Ярославской области разрешили работу бань, но без посетителей",
+    "categorize_entailment: Женщину доставили в больницу, за ее жизнь сейчас борются врачи.",
     "search_query: Сколько программистов нужно, чтобы вкрутить лампочку?",
     # 
-    "paraphrase: What a time to be alive!",
-    "categorize_entailment: Ярославским баням разрешили работать без посетителей",
-    "search_document: Чтобы вкрутить лампочку, требуется три программиста: один напишет программу извлечения лампочки, другой — вкручивания лампочки, а третий проведет тестирование.",
-]
+    "paraphrase: Ярославским баням разрешили работать без посетителей",
+    "categorize_entailment: Женщину спасают врачи.",
+    "search_document: Чтобы вкрутить лампочку, требуется три программиста: один напишет программу извлечения лампочки, другой — вкручивания лампочки, а третий проведет тестирование."
+    ]
 
 tokenizer = AutoTokenizer.from_pretrained("ai-forever/FRIDA")
 model = T5EncoderModel.from_pretrained("ai-forever/FRIDA")
@@ -87,14 +87,14 @@ from sentence_transformers import SentenceTransformer
 
 inputs = [
     # 
-    "paraphrase: Он нам и <unk> не нужон ваш Интернет!",
-    "categorize_entailment: В Ярославской области разрешили работу бань, но без посетителей",
+    "paraphrase: В Ярославской области разрешили работу бань, но без посетителей",
+    "categorize_entailment: Женщину доставили в больницу, за ее жизнь сейчас борются врачи.",
     "search_query: Сколько программистов нужно, чтобы вкрутить лампочку?",
     # 
-    "paraphrase: What a time to be alive!",
-    "categorize_entailment: Ярославским баням разрешили работать без посетителей",
-    "search_document: Чтобы вкрутить лампочку, требуется три программиста: один напишет программу извлечения лампочки, другой — вкручивания лампочки, а третий проведет тестирование.",
-]
+    "paraphrase: Ярославским баням разрешили работать без посетителей",
+    "categorize_entailment: Женщину спасают врачи.",
+    "search_document: Чтобы вкрутить лампочку, требуется три программиста: один напишет программу извлечения лампочки, другой — вкручивания лампочки, а третий проведет тестирование."
+    ]
 
 # loads model with CLS pooling
 model = SentenceTransformer("ai-forever/FRIDA")
@@ -115,11 +115,11 @@ from sentence_transformers import SentenceTransformer
 # loads model with CLS pooling
 model = SentenceTransformer("ai-forever/FRIDA")
 
-classification = model.encode(["Он нам и <unk> не нужон ваш Интернет!", "What a time to be alive!"], prompt_name="paraphrase")
-print(classification[0] @ classification[1].T) # 0.47968706488609314
+paraphrase = model.encode(["В Ярославской области разрешили работу бань, но без посетителей", "Ярославским баням разрешили работать без посетителей"], prompt_name="paraphrase")
+print(paraphrase[0] @ paraphrase[1].T) # 0.47968706488609314
 
-clustering = model.encode(["В Ярославской области разрешили работу бань, но без посетителей", "Ярославским баням разрешили работать без посетителей"], prompt_name="categorize_entailment")
-print(clustering[0] @ clustering[1].T) # 0.940900444984436
+categorize_entailment = model.encode(["Женщину доставили в больницу, за ее жизнь сейчас борются врачи.", "Женщину спасают врачи."], prompt_name="categorize_entailment")
+print(categorize_entailment[0] @ categorize_entailment[1].T) # 0.940900444984436
 
 query_embedding = model.encode("Сколько программистов нужно, чтобы вкрутить лампочку?", prompt_name="search_query")
 document_embedding = model.encode("Чтобы вкрутить лампочку, требуется три программиста: один напишет программу извлечения лампочки, другой — вкручивания лампочки, а третий проведет тестирование.", prompt_name="search_document")
@@ -129,7 +129,7 @@ print(query_embedding @ document_embedding.T) # 0.7761018872261047
 + # Authors
 + [SaluteDevices](https://sberdevices.ru/) AI for B2C RnD Team.
 + Artem Snegirev: [HF profile](https://huggingface.co/artemsnegirev);
-+ Ann Maksimova;
++ Anna Maksimova [HF profile](https://huggingface.co/anpalmak);
 + Aleksandr Abramov: [HF profile](https://huggingface.co/Andrilko), [Github](https://github.com/Ab1992ao), [Kaggle Competitions Master](https://www.kaggle.com/andrilko)