Update README.md

README.md

@@ -1,10 +1,3 @@
-# Style-aware Discriminator
-
-Pretrained weights for [A Style-aware Discriminator for Controllable Image Translation](https://arxiv.org/abs/2203.15375).
-
-Please check the [official repository](https://github.com/kunheek/style-aware-discriminator) for more details.
-
-
 ---
 language:
   - en
@@ -20,4 +13,22 @@ datasets:
   - LSUN churches outdoor
 metrics:
   - fid
----
+---
+
+# Style-aware Discriminator
+
+Pretrained weights for [A Style-aware Discriminator for Controllable Image Translation](https://arxiv.org/abs/2203.15375).
+
+Please check the [official repository](https://github.com/kunheek/style-aware-discriminator) for more details.
+
+
+# Citation
+```sh
+@InProceedings{kim2022style,
+  title={A Style-Aware Discriminator for Controllable Image Translation},
+  author={Kim, Kunhee and Park, Sanghun and Jeon, Eunyeong and Kim, Taehun and Kim, Daijin},
+  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
+  year={2022},
+  pages={18239--18248}
+}
+```