Update README.md

README.md

@@ -21,9 +21,15 @@ 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-base-detox'
+model_name = 's-nlp/bart-base-detox'
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
+
+input_ids = tokenizer.encode('You are a complete idiot!', return_tensors='pt')
+output_ids = model.generate(input_ids, max_length=50, num_return_sequences=1)
+output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+print(output_text)
+# You are unwise!
 ```
 
 **Citation**