Update README.md

README.md

@@ -2,6 +2,7 @@
 language: en
 tags:
 - pythae
+- reproducibility
 license: apache-2.0
 ---
 
@@ -9,5 +10,13 @@ license: apache-2.0
 This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
 ```python
 >>> from pythae.models import AutoModel
->>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
+>>> model = AutoModel.load_from_hf_hub(hf_hub_path="clementchadebec/reproduced_wae")
 ```
+## Reproducibility
+This trained model reproduces the results of Table 1 in [1].
+
+| Model | Dataset | Metric | Obtained value | Reference value |
+|:---:|:---:|:---:|:---:|:---:|
+| WAE | CELEBA 64 | FID | 56.5 | 55 |
+
+[1] Tolstikhin, O Bousquet, S Gelly, and B Schölkopf. Wasserstein auto-encoders. In 6th International Conference on Learning Representations (ICLR 2018), 2018.