dardem commited on
Commit
dba18b4
1 Parent(s): 5bdc74c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -0
README.md CHANGED
@@ -22,3 +22,28 @@ The metrics obtained from test dataset is as follows
22
  | macro avg | 0.96 | 0.96 | 0.96 | 26270 |
23
  | weighted avg | 0.97 | 0.97 | 0.97 | 26270 |
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  | macro avg | 0.96 | 0.96 | 0.96 | 26270 |
23
  | weighted avg | 0.97 | 0.97 | 0.97 | 26270 |
24
 
25
+
26
+ ## How to use
27
+ ```python
28
+ from transformers import BertTokenizer, BertForSequenceClassification
29
+
30
+ # load tokenizer and model weights
31
+ tokenizer = BertTokenizer.from_pretrained('SkolkovoInstitute/russian_toxicity_classifier')
32
+ model = BertForSequenceClassification.from_pretrained('SkolkovoInstitute/russian_toxicity_classifier')
33
+
34
+ # prepare the input
35
+ batch = tokenizer.encode('ты супер', return_tensors='pt')
36
+
37
+ # inference
38
+ model(batch)
39
+ ```
40
+
41
+
42
+ ## Licensing Information
43
+
44
+ [Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License][cc-by-nc-sa].
45
+
46
+ [![CC BY-NC-SA 4.0][cc-by-nc-sa-image]][cc-by-nc-sa]
47
+
48
+ [cc-by-nc-sa]: http://creativecommons.org/licenses/by-nc-sa/4.0/
49
+ [cc-by-nc-sa-image]: https://i.creativecommons.org/l/by-nc-sa/4.0/88x31.png