dardem commited on
Commit
aecf2fe
·
verified ·
1 Parent(s): 92d6598

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -2
README.md CHANGED
@@ -28,8 +28,8 @@ The metrics obtained from test dataset is as follows
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')
@@ -38,6 +38,15 @@ batch = tokenizer.encode('ты супер', return_tensors='pt')
38
  model(batch)
39
  ```
40
 
 
 
 
 
 
 
 
 
 
41
 
42
  ## Licensing Information
43
 
 
28
  from transformers import BertTokenizer, BertForSequenceClassification
29
 
30
  # load tokenizer and model weights
31
+ tokenizer = BertTokenizer.from_pretrained('s-nlp/russian_toxicity_classifier')
32
+ model = BertForSequenceClassification.from_pretrained('s-nlp/russian_toxicity_classifier')
33
 
34
  # prepare the input
35
  batch = tokenizer.encode('ты супер', return_tensors='pt')
 
38
  model(batch)
39
  ```
40
 
41
+ ## Citation
42
+
43
+ ```
44
+ @article{dementieva2022russe,
45
+ title={RUSSE-2022: Findings of the First Russian Detoxification Shared Task Based on Parallel Corpora},
46
+ author={Dementieva, Daryna and Logacheva, Varvara and Nikishina, Irina and Fenogenova, Alena and Dale, David and Krotova, Irina and Semenov, Nikita and Shavrina, Tatiana and Panchenko, Alexander}
47
+ }
48
+ ```
49
+
50
 
51
  ## Licensing Information
52