Update README.md

README.md

@@ -7,7 +7,7 @@ tags:
 
 # BandTok
 
-Minimal standalone inference package for BandTok LM text-to-music generation.
+BandTok is a 2D audio tokenizer that represents music as a time-frequency image. This repository provides the BandTok tokenizer and a language model trained on BandTok tokens for generating 10-second music clips at 44.1 kHz from text prompts.
 
 ## Links
 
@@ -86,3 +86,16 @@ audio = tokenizer.decode(tokens)
 
 - BigVGAN import error: run `git clone https://github.com/NVIDIA/BigVGAN` under /bandtok.
 - T5 download errors: the prompt encoder uses `t5-base`; make sure Hugging Face downloads are available or pre-cache the model.
+
+## Citation
+
+If you find this work useful, please cite:
+
+```bibtex
+@inproceedings{cheng2026modeling,
+  title     = {Modeling Music as a Time-Frequency Image: A 2D Tokenizer for Music Generation},
+  author    = {Cheng, Yuqing and Ma, Xingyu and Yu, Guochen and Gu, Xiaotao},
+  booktitle = {IEEE ICME 2026 Challenge Papers},
+  year      = {2026}
+}
+```