Update README.md

README.md

@@ -1,15 +1,31 @@
 ---
-language:
-- ru
-- en
 tags:
-- translation
-license: apache-2.0
-datasets:
-- wmt19
+- face-recognition
+- face-generation
+- face-segmentation
+- generative-adversarial-network
 metrics:
-- bleu
-- sacrebleu
+- L1
+- PSNR
+- SSIM
+- FID
+datasets:
+- celeba-hq
 ---
+# AOT-GAN CelebA-HQ
+AOT-GAN is a model that can be used for image in-painting. The CelebA-HQ checkpoint is trained on synthetic human faces, which should make it suitable for touching up and restoring portraits.
+
+This model was generated using [AOT-GAN-for-Inpainting](https://github.com/researchmm/AOT-GAN-for-Inpainting), cited as
+
+```
+@inproceedings{yan2021agg,
+  author = {Zeng, Yanhong and Fu, Jianlong and Chao, Hongyang and Guo, Baining},
+  title = {Aggregated Contextual Transformations for High-Resolution Image Inpainting},
+  booktitle = {Arxiv},
+  pages={-},
+  year = {2020}
+}
+```
 
-# Some text here
+## Dataset
+The CelebA-HQ dataset was created with this codebase: https://github.com/tkarras/progressive_growing_of_gans, owned by NVidia and licensed under Creative Commons Attribution-NonCommercial 4.0 International.