Update README.md

README.md

@@ -1,7 +1,14 @@
+---
+language: zh
+tags:
+- wobert
+inference: false
+---
 ## 介绍
-tf版本 https://github.com/ZhuiyiTechnology/WoBERT
-pytorch版本 https://github.com/JunnYu/WoBERT_pytorch
-
+### tf版本 
+https://github.com/ZhuiyiTechnology/WoBERT
+### pytorch版本 
+https://github.com/JunnYu/WoBERT_pytorch
 ## 使用
 ```python
 git clone https://github.com/JunnYu/WoBERT_pytorch
@@ -9,7 +16,6 @@ cd WoBERT_pytorch
 import torch
 from transformers import BertForMaskedLM as WoBertForMaskedLM
 from tokenization_wobert import WoBertTokenizer
-
 pretrained_model_or_path_list = [
     "junnyu/wobert_chinese_plus_base", "junnyu/wobert_chinese_base"
 ]
@@ -32,4 +38,13 @@ for path in pretrained_model_or_path_list:
     print(outputs_sentence)
 # PLUS WoBERT 今天[天气||阳光||天||心情||空气]很好，我[想||要||打算||准备||就]去公园玩。
 # WoBERT 今天[天气||阳光||天||心情||空气]很好，我[想||要||就||准备||也]去公园玩。
-```
+```
+## 引用
+Bibtex：
+```tex
+@techreport{zhuiyiwobert,
+  title={WoBERT: Word-based Chinese BERT model - ZhuiyiAI},
+  author={Jianlin Su},
+  year={2020},
+  url="https://github.com/ZhuiyiTechnology/WoBERT",
+}