s-nlp
/

bart-base-detox

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

dardem commited on May 18, 2022

Commit

d6ef577

·

1 Parent(s): 9670a72

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ The model itself is [BART (base)](https://huggingface.co/facebook/bart-base) mod
 ```python
 from transformers import BartForConditionalGeneration, AutoTokenizer
 base_model_name = 'facebook/bart-base'
-model_name = 'SkolkovoInstitute/bart-detox'
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 ```

 ```python
 from transformers import BartForConditionalGeneration, AutoTokenizer
 base_model_name = 'facebook/bart-base'
+model_name = 'SkolkovoInstitute/bart-base-detox'
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 ```