Update README.md

README.md

@@ -11,11 +11,18 @@ widget:
 
 ## Model description
 
-This model is pre-trained by [UER-py](https://arxiv.org/abs/1909.05658).
+This model is pre-trained by [UER-py](https://github.com/dbiir/UER-py/), which is introduced in [this paper](https://arxiv.org/abs/1909.05658).
+
+You can download the set of Chinese PEGASUS models either from the [UER-py Modelzoo page](https://github.com/dbiir/UER-py/wiki/Modelzoo), or via HuggingFace from the links below:
+
+|                   |              Link              |
+| ----------------- | :----------------------------: |
+| **PEGASUS-Base** | [**L=12/H=768 (Small)**][base] |
+| **PEGASUS-Large**  | [**L=16/H=1024 (Base)**][large]  |
 
 ## How to use
 
-You can use this model directly with a pipeline for text2text generation :
+You can use this model directly with a pipeline for text2text generation (take the case of PEGASUS-Base):
 
 ```python
 >>> from transformers import BertTokenizer, PegasusForConditionalGeneration, Text2TextGenerationPipeline
@@ -28,7 +35,7 @@ You can use this model directly with a pipeline for text2text generation :
 
 ## Training data
 
-[CLUECorpusSmall](https://github.com/CLUEbenchmark/CLUECorpus2020/) is used as training data. 
+[CLUECorpusSmall](https://github.com/CLUEbenchmark/CLUECorpus2020/) is used as training data.
 
 ## Training procedure
 
@@ -81,4 +88,7 @@ python3 scripts/convert_pegasus_from_uer_to_huggingface.py --input_model_path cl
   pages={241},
   year={2019}
 }
-```
+```
+
+[base]:https://huggingface.co/uer/pegasus-base-chinese-cluecorpussmall
+[large]:https://huggingface.co/uer/pegasus-large-chinese-cluecorpussmall