update

README.md

@@ -32,7 +32,7 @@ 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使用
@@ -40,8 +40,8 @@ print(pt_outputs_sentence)
 import tensorflow as tf
 from transformers import RoFormerTokenizer, TFRoFormerForMaskedLM
 text = "今天[MASK]很好，我[MASK]去公园玩。"
-tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_base")
-tf_model = TFRoFormerForMaskedLM.from_pretrained("junnyu/roformer_chinese_base")
+tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_char_base")
+tf_model = TFRoFormerForMaskedLM.from_pretrained("junnyu/roformer_chinese_char_base")
 tf_inputs = tokenizer(text, return_tensors="tf")
 tf_outputs = tf_model(**tf_inputs, training=False).logits[0]
 tf_outputs_sentence = "tf2.0: "
@@ -54,7 +54,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 今天[天||气||都||风||人]很好，我[想||要||就||也||还]去公园玩。
 ```
 
 ## 引用