Update README.md

README.md

@@ -1,33 +1,38 @@
-# gpt2-chinese-ai-text-classifier
-
-使用 `uer/gpt2-chinese-cluecorpussmall` 作为基座模型，基于 [Hello-SimpleAI/HC3-Chinese](https://huggingface.co/datasets/Hello-SimpleAI/HC3-Chinese) 数据集训练得到的 AI 文本分类器。
-
-## 训练数据
-
-[Hello-SimpleAI/HC3-Chinese](https://huggingface.co/datasets/Hello-SimpleAI/HC3-Chinese)
-
-## Demo
-
-- gradio==3.50.0
-
-- easytokenizer
-
-- transformers
-
-安装 [easytokenizer](https://github.com/zejunwang1/easytokenizer)：
-
-```shell
-pip install git+https://github.com/zejunwang1/easytokenizer
-```
-
-启动 demo：
-
-```shell
-python gradio_demo.py --model gpt2-chinese-ai-text-classifier/ --port 9000
-```
-
-打开 `http://[server_ip]:9000/`，查看效果：
-
-![](D:\nlp\doc\markdown\gpt_detector_1.png)
-
-![](D:\nlp\doc\markdown\gpt_detector_2.png)
+---
+license: apache-2.0
+base_model:
+- uer/gpt2-chinese-cluecorpussmall
+---
+# gpt2-chinese-ai-text-classifier
+
+使用 `uer/gpt2-chinese-cluecorpussmall` 作为基座模型，基于 [Hello-SimpleAI/HC3-Chinese](https://huggingface.co/datasets/Hello-SimpleAI/HC3-Chinese) 数据集训练得到的 AI 文本分类器。
+
+## 训练数据
+
+[Hello-SimpleAI/HC3-Chinese](https://huggingface.co/datasets/Hello-SimpleAI/HC3-Chinese)
+
+## Demo
+
+- gradio==3.50.0
+
+- easytokenizer
+
+- transformers
+
+安装 [easytokenizer](https://github.com/zejunwang1/easytokenizer)：
+
+```shell
+pip install git+https://github.com/zejunwang1/easytokenizer
+```
+
+启动 demo：
+
+```shell
+python gradio_demo.py --model gpt2-chinese-ai-text-classifier/ --port 9000
+```
+
+打开 `http://[server_ip]:9000/`，查看效果：
+
+![](D:\nlp\doc\markdown\gpt_detector_1.png)
+
+![](D:\nlp\doc\markdown\gpt_detector_2.png)