Update README.md
Browse files
README.md
CHANGED
|
@@ -1,11 +1,13 @@
|
|
| 1 |
---
|
| 2 |
datasets:
|
| 3 |
- s-nlp/paradetox
|
|
|
|
| 4 |
language:
|
| 5 |
- ru
|
| 6 |
- en
|
| 7 |
library_name: transformers
|
| 8 |
pipeline_tag: text2text-generation
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
## Model Description
|
|
@@ -21,7 +23,7 @@ The model is based on [mBART-large-50](https://huggingface.co/facebook/mbart-lar
|
|
| 21 |
```python
|
| 22 |
from transformers import MBartForConditionalGeneration, AutoTokenizer
|
| 23 |
|
| 24 |
-
model = MBartForConditionalGeneration.from_pretrained("s-nlp/
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50")
|
| 26 |
|
| 27 |
```
|
|
|
|
| 1 |
---
|
| 2 |
datasets:
|
| 3 |
- s-nlp/paradetox
|
| 4 |
+
- s-nlp/ru_paradetox
|
| 5 |
language:
|
| 6 |
- ru
|
| 7 |
- en
|
| 8 |
library_name: transformers
|
| 9 |
pipeline_tag: text2text-generation
|
| 10 |
+
license: openrail++
|
| 11 |
---
|
| 12 |
|
| 13 |
## Model Description
|
|
|
|
| 23 |
```python
|
| 24 |
from transformers import MBartForConditionalGeneration, AutoTokenizer
|
| 25 |
|
| 26 |
+
model = MBartForConditionalGeneration.from_pretrained("s-nlp/mbart-detox-en-ru").cuda()
|
| 27 |
tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50")
|
| 28 |
|
| 29 |
```
|