Update README.md

README.md

@@ -26,28 +26,20 @@ RougeL= 23.499
 ## Intended uses & limitations
 To use its API:
 
-
-from transformers import BertTokenizerFast, GPT2Tokenizer, EncoderDecoderModel
-
+from transformers import RobertaTokenizerFast, GPT2Tokenizer, EncoderDecoderModel
 
 model = EncoderDecoderModel.from_pretrained("Ayham/roberta_gpt2_summarization_cnn_dailymail")
 
+input_tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')
 
-input_tokenizer = BertTokenizerFast.from_pretrained('bert-base-cased')
-
+output_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 article = """Your Input Text"""
 
-
 input_ids = input_tokenizer(article, return_tensors="pt").input_ids
 
-
 output_ids = model.generate(input_ids)
 
-
-output_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-
-
 print(output_tokenizer.decode(output_ids[0], skip_special_tokens=True))
 
 More information needed