Add pipeline tag and GitHub link

README.md

@@ -1,20 +1,26 @@
 ---
-library_name: transformers
-license: mit
 base_model:
 - openai-community/gpt2
+library_name: transformers
+license: mit
+pipeline_tag: text-generation
 ---
+
 # CODI Model
 
 <div align="center">
 
 [![HuggingFace](https://img.shields.io/badge/🤗%20HuggingFace-Model-fcc21b?style=for-the-badge&logo=huggingface&logoColor=white)](https://huggingface.co/ModalityDance/latent-tts-codi)
+[![Paper](https://img.shields.io/badge/Paper-arXiv-b31b1b?style=for-the-badge&logo=arxiv)](https://huggingface.co/papers/2510.07745)
+[![GitHub](https://img.shields.io/badge/GitHub-Code-black?style=for-the-badge&logo=github)](https://github.com/ModalityDance/LatentTTS)
 
 </div>
 
 ## Overview
 
-**CODI** (Continuous Chain-of-Thought via Self-Distillation) is a latent reasoning model based on GPT-2 that extends the base architecture with an optional projector module for enhanced hidden state representations. This model is part of the [Parallel Test-Time Scaling for Latent Reasoning Models](https://arxiv.org/abs/2510.07745) framework.
+**CODI** (Continuous Chain-of-Thought via Self-Distillation) is a latent reasoning model based on GPT-2 that extends the base architecture with an optional projector module for enhanced hidden state representations. This model is part of the [Parallel Test-Time Scaling for Latent Reasoning Models](https://huggingface.co/papers/2510.07745) framework.
+
+The official implementation is available at [github.com/ModalityDance/LatentTTS](https://github.com/ModalityDance/LatentTTS).
 
 ## Model Details
 
@@ -184,7 +190,7 @@ from src.paths import extract_answer_number
 
 # Extract answer from generated text
 answer = extract_answer_number(result)
-print(f"Answer: {answer}")
+print(f\"Answer: {answer}\")
 ```
 
 ## Evaluation
@@ -196,12 +202,6 @@ Run evaluation using the provided scripts:
 ./run_tests.sh
 ```
 
-## Model Card
-
-- **Paper**: [Parallel Test-Time Scaling for Latent Reasoning Models](https://arxiv.org/abs/2510.07745)
-- **HuggingFace**: [ModalityDance/latent-tts-codi](https://huggingface.co/ModalityDance/latent-tts-codi)
-- **Benchmarks**: GSM8K Test, GSM8K Hard, MultiArith
-
 ## Citation
 
 If you use this model, please cite: