Update README.md
Browse files
README.md
CHANGED
@@ -34,10 +34,10 @@ The preferable usage:
|
|
34 |
# pip install transformers
|
35 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
36 |
import torch
|
37 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
38 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
39 |
# model.cuda()
|
40 |
-
inputs = tokenizer('привет[SEP]привет![SEP]как дела?[RESPONSE_TOKEN]
|
41 |
padding=True, max_length=128, truncation=True, add_special_tokens=False, return_tensors='pt')
|
42 |
with torch.inference_mode():
|
43 |
logits = model(**inputs).logits
|
|
|
34 |
# pip install transformers
|
35 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
36 |
import torch
|
37 |
+
tokenizer = AutoTokenizer.from_pretrained("tinkoff-ai/response-quality-classifier-tiny")
|
38 |
+
model = AutoModelForSequenceClassification.from_pretrained("tinkoff-ai/response-quality-classifier-tiny")
|
39 |
# model.cuda()
|
40 |
+
inputs = tokenizer('привет[SEP]привет![SEP]как дела?[RESPONSE_TOKEN]норм, у тя как?',
|
41 |
padding=True, max_length=128, truncation=True, add_special_tokens=False, return_tensors='pt')
|
42 |
with torch.inference_mode():
|
43 |
logits = model(**inputs).logits
|