Update README.md

README.md

@@ -2,13 +2,13 @@
 language: "fr"
 ---
 
-# BelGPT-2
+# Belgian GPT-2 🇧🇪
 
-**BelGPT-2** (*Belgian GPT-2* 🇧🇪) is a "small" GPT-2 model pre-trained on a very large and heterogeneous French corpus (around 60Gb). Please check [antoiloui/gpt2-french](https://github.com/antoiloui/gpt2-french) for more information about the pre-trained model, its implementation, and the data.
+**A GPT-2 model pre-trained on a very large and heterogeneous French corpus (~60Gb).**
 
-## Using BelGPT-2 for Text Generation in French
+## Usage
 
-You can use BelGPT-2 with [🤗 transformers](https://github.com/huggingface/transformers) library as follows:
+You can use BelGPT-2 with [🤗 transformers](https://github.com/huggingface/transformers):
 
 ```python
 import torch
@@ -50,3 +50,20 @@ Below is the list of all French copora used to pre-trained the model:
 | EuroParl |  `europarl`   |  289.9 MB   |   278.7 MB  |
 | NewsCommentary |  `news_commentary`   |   61.4 MB  |  58.1 MB   |
 | **Total** |     |   **236.3 GB**  |  **57.9 GB**   |
+
+## Documentation
+
+Detailed documentation on the pre-trained model, its implementation, and the data can be found [here](https://github.com/antoiloui/belgpt2/blob/master/docs/index.md).
+
+## Citation
+
+For attribution in academic contexts, please cite this work as:
+
+```
+@misc{louis2020belgpt2,
+  author = {Louis, Antoine},
+  title = {{BelGPT-2: a GPT-2 model pre-trained on French corpora.}},
+  year = {2020},
+  howpublished = {\url{https://github.com/antoiloui/belgpt2}},
+}
+```