Update README.md

README.md

@@ -21,19 +21,20 @@ widget:
 # TeraSpace/replica_classification
 Сделано на основе [xlm-roberta-base](https://huggingface.co/xlm-roberta-base)
 
-1. about_user - реагирует, когда пользователь говорит о себе. Например, "меня зовут андрей"
-2. quest - реагирует на вопрос не относящийся к диалогу, например: "когда родился пушкин" или "когда я стану миллионером"
-3. instruct - реагирует на вопросы, ответ на которые представляет собой инструкцию. Например, "как установить windows, как приготовить борщ"
+0. dialog - реагирует на диалоговые реплики. Например, "привет"
+1. trouble - реагирует на реплики, где пользователь рассказывает о своих проблемах. Например, "у меня болит зуб, мне проткнули колесо"
+2. question - реагирует на вопрос не относящийся к диалогу, например: "когда родился пушкин" или "когда я стану миллионером"
+3. about_user - реагирует, когда пользователь говорит о себе. Например, "меня зовут андрей"
 4. about_model - реагирует на вопросы о личности ассистента. Например, "как тебя зовут, ты кто такая"
-5. trouble - реагирует на реплики, где пользователь рассказывает о своих проблемах. Например, "у меня болит зуб, мне проткнули колесо"
-7. dialog - реагирует на диалоговые реплики. Например, "привет"
+5. instruct - реагирует на вопросы, ответ на которые представляет собой инструкцию. Например, "как установить windows, как приготовить борщ"
+
 # Использование
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 del_symbs = ["?","!",".",","]
-classes = ["dialog","trouble","quest","about_user","about_model","instruction"]
+classes = ["dialog","trouble","question","about_user","about_model","instruct"]
 
 device = torch.device("cuda")
 model_name = 'TeraSpace/replica_classification'