Update README.md

README.md

@@ -18,6 +18,7 @@ widget:
   - text: 'меня артур зовут'
     example_title: example_7
 ---
+# Den4ikAI/ruBert-tiny-replicas-classifier
 Описание классов:
 1. about_user - реагирует, когда пользователь говорит о себе. Например, "меня зовут андрей"
 2. inaccurate_question - реагирует на философские и размытые вопросы. Например, "что такое любовь, когда деньги придут за мной"
@@ -29,15 +30,15 @@ widget:
 
 Примечание: модель обучалась без знаков '?'
 
-Код инференса:
+# Использование
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/rubert_tiny_questions_classifier')
-model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/rubert_tiny_questions_classifier')
+tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/ruBert-tiny-replicas-classifier')
+model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-tiny-replicas-classifier')
 model.to(device)
 model.eval()