Den4ikAI
/

ruBert-tiny-replicas-classifier

Text Classification

text-embeddings-inference

Model card Files Files and versions

Den4ikAI commited on Jul 8, 2023

Commit

267363f

·

1 Parent(s): 33024c5

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-tiny
 model.to(device)
 model.eval()
-classes = ['about_user', 'inaccurate_question', 'exact_question', 'instruct', 'about_system', 'problem', 'dialogue']
 def get_sentence_type(text):
@@ -54,4 +54,5 @@ def get_sentence_type(text):
 while 1:
   print(get_sentence_type(input(":> ")))
 ```

 model.to(device)
 model.eval()
+classes = ['instruct', 'question', 'dialogue', 'problem', 'about_system', 'about_user']
 def get_sentence_type(text):
 while 1:
   print(get_sentence_type(input(":> ")))
 ```