Den4ikAI commited on
Commit
91bbc9e
1 Parent(s): ec8f79c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -1
README.md CHANGED
@@ -6,5 +6,29 @@ language:
6
  - ru
7
  pipeline_tag: text-classification
8
  ---
9
- #
10
  Модель для оценки релевантности ответов на вопросы.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  - ru
7
  pipeline_tag: text-classification
8
  ---
9
+ # Den4ikAI/ruBert-base-qa-ranker
10
  Модель для оценки релевантности ответов на вопросы.
11
+ # Использование
12
+ ```python
13
+ import torch
14
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
15
+
16
+ tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/ruBert-base-qa-ranker')
17
+ model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-base-qa-ranker')
18
+ inputs = tokenizer('[CLS]Что такое QR-код?[RESPONSE_TOKEN]QR-код - это тип матричного штрих-кода.', max_length=512, add_special_tokens=False, return_tensors='pt')
19
+ with torch.inference_mode():
20
+ logits = model(**inputs).logits
21
+ probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
22
+ relevance, no_relevance = probas
23
+ print('Relevance: {}'.format(relevance))
24
+ ```
25
+ # Citation
26
+ ```
27
+ @MISC{Den4ikAI/ruBert-base-qa-ranker,
28
+ author = {Denis Petrov},
29
+ title = {Russian QA relevancy model},
30
+ url = {https://huggingface.co/Den4ikAI/ruBert-base-qa-ranker},
31
+ year = 2023
32
+ }
33
+
34
+ ```