docs: fix usage and add dataset citation

README.md

@@ -108,7 +108,7 @@ print(label2id)
 sentence = "黄龙国际中心位于杭州市西湖区学院路77号，地处杭州黄龙商务区核心位置。项目以“新经济、新生态、新黄龙”的理念和“知识城市创新区”的项目定位，引领着杭州城市的产业升级和创新。\n黄龙国际中心主打“知识盒子”建筑设计，拥有时尚、新潮的建筑立面，聚集不同功能的独立建筑物，打造出包容开放又具有链接性的空间。项目在2018年4月一举斩获开发项目行业最高奖——广厦奖。\n项目整体分四期开发。一期G、H、I三幢楼宇由坤和自主开发建成，于2015年4月投入使用，于2015年5月获得了美国绿色建筑委员会颁发的LEED金级认证，并成功吸引蚂蚁金服、美图、有赞等国内互联网创新巨头率先入驻。\n2016年初，坤和引进万科作为该项目股权合作方通过双方优质资源共享，在产品品质、创新等方面得到全方位提升。\n二期A、B幢由美国KPF设计，并于2018年4月一举获得房地产开发项目行业最高奖——广厦奖。A、B幢写字楼于2018年10月投入使用，B幢与全球领先的创造者社区及空间和服务提供商——WeWork正式签约。商业K-lab于2018年12月28日正式开业。\n项目三期E、F幢已于2020年投入使用。四期C、D幢，计划于2021年底投入使用。\n项目总占地面积约8.7万方，总建筑面积约40万平方米，涵盖9栋国际5A写字楼及8万方K-lab商业新物种，以知识为纽带，打造成一个集商务、商业、教育、文化、娱乐、艺术、餐饮、会展等于一体的完整城市性生态体系。项目全部投入使用后，年租金收入将超6亿元。\n"
 
 result = tokenizer(
-            [sentecnce],
+            [sentence],
             padding=False,
             max_length=2048,
             truncation=True,
@@ -119,4 +119,19 @@ result["input_ids"].to(dtype=torch.bfloat16)
 model_out = model(**result)
 predict_label_id = torch.argmax(model_out.logits, dim=-1).tolist()[0]
 predict_label_str = id2label[predict_label_id]
-```
+```
+
+## Citation
+
+If you use this classifier or IndustryCorpus2, please cite the source dataset:
+
+```bibtex
+@misc{shi2024industrycorpus2,
+  title     = {IndustryCorpus2},
+  author    = {Xiaofeng Shi and Lulu Zhao and Hua Zhou and Donglin Hao},
+  year      = {2024},
+  publisher = {Hugging Face},
+  doi       = {10.57967/hf/3488},
+  url       = {https://huggingface.co/datasets/BAAI/IndustryCorpus2}
+}
+```