Update README.md

README.md

@@ -1,14 +1,15 @@
 ---
 language:
-  - en
-thumbnail: "url to a thumbnail used in social sharing"
+- en
+thumbnail: url to a thumbnail used in social sharing
 tags:
-  - ArguGPT
+- ArguGPT
 license: mit
 datasets:
-  - ArguGPT
+- ArguGPT
 metrics:
-  - accuracy
+- accuracy
+pipeline_tag: text-classification
 ---
 
 # ArguGPT
@@ -19,3 +20,18 @@ RoBERTa-large finetuned on ArguGPT essays.
 - label 0 for human written essays
 
 **Please truncate your input essay to 512 tokens**
+
+## Citation
+
+Please cite our work [arXiv:2304.07666](https://arxiv.org/abs/2304.07666) as  
+
+```
+@misc{liu2023argugpt,
+      title={ArguGPT: evaluating, understanding and identifying argumentative essays generated by GPT models}, 
+      author={Yikang Liu and Ziyin Zhang and Wanyang Zhang and Shisen Yue and Xiaojing Zhao and Xinyuan Cheng and Yiwen Zhang and Hai Hu},
+      year={2023},
+      eprint={2304.07666},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```