Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-tiny
|
|
41 |
model.to(device)
|
42 |
model.eval()
|
43 |
|
44 |
-
classes = ['
|
45 |
|
46 |
|
47 |
def get_sentence_type(text):
|
@@ -54,4 +54,5 @@ def get_sentence_type(text):
|
|
54 |
|
55 |
while 1:
|
56 |
print(get_sentence_type(input(":> ")))
|
|
|
57 |
```
|
|
|
41 |
model.to(device)
|
42 |
model.eval()
|
43 |
|
44 |
+
classes = ['instruct', 'question', 'dialogue', 'problem', 'about_system', 'about_user']
|
45 |
|
46 |
|
47 |
def get_sentence_type(text):
|
|
|
54 |
|
55 |
while 1:
|
56 |
print(get_sentence_type(input(":> ")))
|
57 |
+
|
58 |
```
|