Update README.md

README.md

@@ -6,24 +6,11 @@ tags:
 - tf2.0
 inference: false
 ---
-## 介绍
-### tf版本 
-https://github.com/ZhuiyiTechnology/roformer
-
-### pytorch版本+tf2.0版本
-https://github.com/JunnYu/RoFormer_pytorch
-
-## 安装
-```bash
-pip install roformer
-或
-pip install git+https://github.com/JunnYu/RoFormer_pytorch.git
-```
 
 ## pytorch使用
 ```python
 import torch
-from roformer import RoFormerForMaskedLM, RoFormerTokenizer
+from transformers import RoFormerForMaskedLM, RoFormerTokenizer
 
 text = "今天[MASK]很好，我[MASK]去公园玩。"
 tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_small")
@@ -47,7 +34,7 @@ print(pt_outputs_sentence)
 ## tensorflow2.0使用
 ```python
 import tensorflow as tf
-from roformer import RoFormerTokenizer, TFRoFormerForMaskedLM
+from transformers import RoFormerTokenizer, TFRoFormerForMaskedLM
 text = "今天[MASK]很好，我[MASK]去公园玩。"
 tokenizer = RoFormerTokenizer.from_pretrained("junnyu/roformer_chinese_small")
 tf_model = TFRoFormerForMaskedLM.from_pretrained("junnyu/roformer_chinese_small")