Update README.md

README.md

@@ -6,7 +6,7 @@ metrics:
 - bleu
 pipeline_tag: translation
 widget:
-- text: "安世字子孺，少以父任为郎"
+- text: "竖子不足与谋。"
   example_title: "translation"
 ---
 # Model Description
@@ -16,14 +16,15 @@ Erya4FT is based on [Erya](https://huggingface.co/RUCAIBox/Erya) and further fin
 # Example
 
 ```python
-from transformers import BertTokenizer, CPTForConditionalGeneration
+>>> from transformers import BertTokenizer, CPTForConditionalGeneration
 
-tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya4FT")
-model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
+>>> tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya4FT")
+>>> model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
 
-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))
+['这 小 子 不 值 得 与 他 商 量 。']
 ```