Update README.md

README.md

@@ -40,17 +40,32 @@ Our language model was generated using a dataset of Italian wikipedia articles a
 
 
 ## Download CommonVoice8.0 dataset for italian language
-
+```python
 from datasets import load_dataset
 
 dataset = load_dataset("mozilla-foundation/common_voice_8_0", "it", use_auth_token=True)
-
+```
 
 ## Evaluation Commands
 
 To evaluate on `mozilla-foundation/common_voice_8_0` with split `test`:
 
+```bash
 python eval.py --model_id radiogroup-crits/wav2vec2-xls-r-1b-italian-doc4lm-5gram --dataset mozilla-foundation/common_voice_8_0 --config it --split test --log_outputs --greedy
 mv log_mozilla-foundation_common_voice_8_0_it_test_predictions.txt log_mozilla-foundation_common_voice_8_0_it_test_predictions_greedy.txt
 mv mozilla-foundation_common_voice_8_0_it_test_eval_results.txt mozilla-foundation_common_voice_8_0_it_test_eval_results_greedy.txt
+```
+
+## Citation
+If you want to cite this model you can use this:
 
+```bibtex
+@misc{radiogroup-crits2022wav2vec2-xls-r-1b-italian-doc4lm-5gram,
+  title={XLS-R Wav2Vec2 Italian by radiogroup-crits},
+  author={Raffaele Teraoni Prioletti and Paolo Casagranda and Francesco Russo},
+  publisher={Hugging Face},
+  journal={Hugging Face Hub},
+  howpublished={\url{https://huggingface.co/radiogroup-crits/wav2vec2-xls-r-1b-italian-doc4lm-5gram}},
+  year={2022}
+}
+```