Update README.md

README.md

@@ -11,6 +11,10 @@ tags:
 - chats
 - embeddings
 - coherence
+widget:
+- text: Где можно получить СНИЛС? Можете в МФЦ
+- text: Я тут уже много лет Куда отправить это письмо?
+pipeline_tag: text-classification
 ---
 # Model Card
 
@@ -26,7 +30,7 @@ The task perfectly aligns with the Next Sentence Prediction task, so the fine-tu
 **Note:** if two messages have `reply_to` relationship, then **they have "zero" label**. This is because of the NSP formulation.
 ```python
 from transformers import AutoTokenizer, BertForNextSentencePrediction
-tokenizer = AutoTokenizer.from_pretrained("rubert_reply_recovery", )
+tokenizer = AutoTokenizer.from_pretrained("astromis/rubert_reply_recovery", )
 model = BertForNextSentencePrediction.from_pretrained("rubert_reply_recovery", )
 
 inputs = tokenizer(['Где можно получить СНИЛС?', 'Я тут уже много лет'], ["Можете в МФЦ", "Куда отправить это письмо?"], return_tensors='pt',