Update README.md

README.md

@@ -19,7 +19,7 @@ https://github.com/JunnYu/RoFormer_pytorch
 import torch
 from transformers import RoFormerForMaskedLM, RoFormerTokenizer
 
-text = "今天[MASK]很好，我[MASK]去公园玩。"
+text = "今天[MASK]很好，我想去公园玩！"
 tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_base")
 pt_model = RoFormerForMaskedLM.from_pretrained("junnyu/roformer_chinese_base")
 pt_inputs = tokenizer(text, return_tensors="pt")
@@ -34,14 +34,14 @@ for i, id in enumerate(tokenizer.encode(text)):
         pt_outputs_sentence += "".join(
             tokenizer.convert_ids_to_tokens([id], skip_special_tokens=True))
 print(pt_outputs_sentence)
-# pytorch 今天[天气||天||心情||阳光||空气]很好，我[想||要||打算||准备||喜欢]去公园玩。
+# pytorch: 今天[天气||天||阳光||太阳||空气]很好，我想去公园玩！
 ```
 
 ## tensorflow2.0使用
 ```python
 import tensorflow as tf
 from transformers import RoFormerTokenizer, TFRoFormerForMaskedLM
-text = "今天[MASK]很好，我[MASK]去公园玩。"
+text = "今天[MASK]很好，我想去公园玩！"
 tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_base")
 tf_model = TFRoFormerForMaskedLM.from_pretrained("junnyu/roformer_chinese_base")
 tf_inputs = tokenizer(text, return_tensors="tf")
@@ -56,7 +56,7 @@ for i, id in enumerate(tokenizer.encode(text)):
         tf_outputs_sentence += "".join(
             tokenizer.convert_ids_to_tokens([id], skip_special_tokens=True))
 print(tf_outputs_sentence)
-# tf2.0 今天[天气||天||心情||阳光||空气]很好，我[想||要||打算||准备||喜欢]去公园玩。
+# tf2.0: 今天[天气||天||阳光||太阳||空气]很好，我想去公园玩！
 ```
 
 ## 引用