Update README.md

README.md

@@ -17,14 +17,14 @@ More information about Erya dataset can be found here: [RUCAIBox/Erya-dataset ·
 # Example
 
 ```python
-from transformers import BertTokenizer, CPTForConditionalGeneration
+>>> from transformers import BertTokenizer, CPTForConditionalGeneration
 
-tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya")
-model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya")
+>>> tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya")
+>>> model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya")
 
-input_ids = tokenizer("安世字子孺，少以父任为郎。", return_tensors='pt')
-input_ids.pop("token_type_ids")
+>>> input_ids = tokenizer("安世字子孺，少以父任为郎。", return_tensors='pt')
+>>> input_ids.pop("token_type_ids")
 
-pred_ids = model.generate(max_new_tokens=256, **input_ids)
-print(tokenizer.batch_decode(pred_ids, skip_special_tokens=True))
+>>> pred_ids = model.generate(max_new_tokens=256, **input_ids)
+>>> print(tokenizer.batch_decode(pred_ids, skip_special_tokens=True))
 ```