Update README.md

README.md

@@ -22,20 +22,36 @@ widget:
 - text: "Question:How should covid-19 be prevented? Answer:"
   example_title: "test Question2"
 ---
-# YuyuanQA-GPT2-3.5B model (Medical)，one model of [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM).
+
+# YuyuanQA-GPT2-3.5B
+
+- Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
+- Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
+
+## 模型分类 Model Taxonomy
+
+|  需求 Demand  | 任务 Task       | 系列 Series      | 模型 Model    | 参数 Parameter | 额外 Extra |
+|  :----:  | :----:  | :----:  | :----:  | :----:  | :----:  |
+| 特殊 Special | 领域 Domain | 余元 Yuyuan | GPT2 |      3.5B      |     问答 QA    |
+
+## 模型信息 Model Information
+
+
+
 **YuyuanQA-GPT2-3.5B** is fine-tuned with 10000 medical QA pairs based on **Yuyuan-3.5B** model.
 
 **Question answering（QA）** is an important subject related to natural language processing and information retrieval. There are many application scenarios in the actual industry. **Traditional methods are often complex**, and their core algorithms involve **machine learning**, **deep learning** and **knowledge graph** related knowledge.
 
-We hope to explore a **simpler** and more **effective** way to use the powerful memory and understanding ability of the large model to directly realize question and answering. Yuyuanqa-GPT2-3.5b model is an attempt and **performs well under subjective test**. At the same time, we also tested 100 QA pairs with ***BLEU***:
+We hope to explore a **simpler** and more **effective** way to use the powerful memory and understanding ability of the large model to directly realize question and answering. Yuyuanqa-GPT2-3.5b model is an attempt and **performs well under subjective test**. At the same time, we also tested 100 QA pairs with **BLEU**:
 
 | gram | 1-gram | 2-gram | 3-gram | 4-gram |
 | ----------- | ----------- |------|------|------|
-| **blue_score**   | 0.357727 | 0.2713 | 0.22304 | 0.19099 |
+| blue_score   | 0.357727 | 0.2713 | 0.22304 | 0.19099 |
+
+## 使用 Usage
 
-## Usage
+### 加载模型 Loading Models
 
-### load model
 ```python 
 from transformers import GPT2Tokenizer,GPT2LMHeadModel
 
@@ -44,7 +60,9 @@ hf_model_path = 'model_path or model name'
 tokenizer = GPT2Tokenizer.from_pretrained(hf_model_path)
 model = GPT2LMHeadModel.from_pretrained(hf_model_path)
 ```
-### generation
+
+### 使用示例 Usage Examples
+
 ```python
 fquestion = "What should gout patients pay attention to in diet?"
 inputs = tokenizer(f'Question:{question} answer:',return_tensors='pt')
@@ -66,19 +84,36 @@ for idx,sentence in enumerate(generation_output.sequences):
     print('*'*40)
 
 ```
-## example
 
 We made a demo of medical Q & A with YuyuanQA-GPT2-3.5B model. In the future, we will make this product into a wechat app to meet you. Please look forward to it.
 
 ![avatar](https://huggingface.co/IDEA-CCNL/YuyuanQA-GPT2-3.5B/resolve/main/QA-DEMO.png)
 
-## Citation
-If you find the resource is useful, please cite the following website in your paper.
+## 引用 Citation
+
+如果您在您的工作中使用了我们的模型，可以引用我们的[论文](https://arxiv.org/abs/2209.02970)：
+
+If you are using the resource for your work, please cite the our [paper](https://arxiv.org/abs/2209.02970):
+
+```text
+@article{fengshenbang,
+  author    = {Junjie Wang and Yuxiang Zhang and Lin Zhang and Ping Yang and Xinyu Gao and Ziwei Wu and Xiaoqun Dong and Junqing He and Jianheng Zhuo and Qi Yang and Yongfeng Huang and Xiayu Li and Yanghan Wu and Junyu Lu and Xinyu Zhu and Weifeng Chen and Ting Han and Kunhao Pan and Rui Wang and Hao Wang and Xiaojun Wu and Zhongshen Zeng and Chongpei Chen and Ruyi Gan and Jiaxing Zhang},
+  title     = {Fengshenbang 1.0: Being the Foundation of Chinese Cognitive Intelligence},
+  journal   = {CoRR},
+  volume    = {abs/2209.02970},
+  year      = {2022}
+}
 ```
+
+也可以引用我们的[网站](https://github.com/IDEA-CCNL/Fengshenbang-LM/):
+
+You can also cite our [website](https://github.com/IDEA-CCNL/Fengshenbang-LM/):
+
+```text
 @misc{Fengshenbang-LM,
   title={Fengshenbang-LM},
   author={IDEA-CCNL},
-  year={2022},
+  year={2021},
   howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
 }
 ```