Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ The model itself is [BART (base)](https://huggingface.co/facebook/bart-base) mod
|
|
8 |
```python
|
9 |
from transformers import BartForConditionalGeneration, AutoTokenizer
|
10 |
base_model_name = 'facebook/bart-base'
|
11 |
-
model_name = 'SkolkovoInstitute/bart-detox'
|
12 |
tokenizer = AutoTokenizer.from_pretrained(base_model_name)
|
13 |
model = BartForConditionalGeneration.from_pretrained(model_name)
|
14 |
```
|
|
|
8 |
```python
|
9 |
from transformers import BartForConditionalGeneration, AutoTokenizer
|
10 |
base_model_name = 'facebook/bart-base'
|
11 |
+
model_name = 'SkolkovoInstitute/bart-base-detox'
|
12 |
tokenizer = AutoTokenizer.from_pretrained(base_model_name)
|
13 |
model = BartForConditionalGeneration.from_pretrained(model_name)
|
14 |
```
|