Den4ikAI commited on
Commit
267363f
·
1 Parent(s): 33024c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -41,7 +41,7 @@ model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-tiny
41
  model.to(device)
42
  model.eval()
43
 
44
- classes = ['about_user', 'inaccurate_question', 'exact_question', 'instruct', 'about_system', 'problem', 'dialogue']
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
  ```