Improve model card for ThinkSound with metadata, detailed content, and links

README.md

@@ -1,21 +1,163 @@
 ---
 license: apache-2.0
+pipeline_tag: other
+tags:
+- audio-generation
+- video-to-audio
+- multimodal
+- chain-of-thought
+- audio-editing
+- foley-generation
 ---
 
-This repository contains the weights of  [ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing](https://arxiv.org/abs/2506.21448).
+# 🎶 ThinkSound
 
-Project Page: https://thinksound-project.github.io/.
+This repository contains the weights of [ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing](https://huggingface.co/papers/2506.21448).
 
-If you find our work useful, please cite our paper: 
+Project Page: https://thinksound-project.github.io/
+
+Code: https://github.com/liuhuadai/ThinkSound
+
+<p align="center">
+  If you find this project useful, a star ⭐ on GitHub would be greatly appreciated!
+</p>
+
+<p align="center">
+  <a href="https://huggingface.co/papers/2506.21448">
+    <img src="https://img.shields.io/badge/Paper-2506.21448-b31b1b.svg" alt="Paper"/>
+  </a>
+  &nbsp;
+  <a href="https://thinksound-project.github.io/">
+    <img src="https://img.shields.io/badge/Online%20Demo-🌐-blue" alt="Online Demo"/>
+  </a>
+  &nbsp;
+  <a href="https://huggingface.co/spaces/FunAudioLLM/ThinkSound">
+    <img src="https://img.shields.io/badge/HuggingFace-Spaces-orange?logo=huggingface" alt="Hugging Face Space"/>
+  </a>
+  &nbsp;
+  <a href="https://modelscope.cn/studios/iic/ThinkSound">
+    <img src="https://img.shields.io/badge/ModelScope-在线体验-green" alt="ModelScope"/>
+  </a>
+</p>
+
+---
+
+**ThinkSound** is a unified Any2Audio generation framework with flow matching guided by Chain-of-Thought (CoT) reasoning.
+
+PyTorch implementation for multimodal audio generation and editing: generate or edit audio from video, text, and audio, powered by step-by-step reasoning from Multimodal Large Language Models (MLLMs).
+
+![Teaser](https://github.com/liuhuadai/ThinkSound/blob/main/assets/figs/fig1_teaser.png)
+
+---
+
+## Abstract
+While end-to-end video-to-audio generation has greatly improved, producing high-fidelity audio that authentically captures the nuances of visual content remains challenging. Like professionals in the creative industries, such generation requires sophisticated reasoning about items such as visual dynamics, acoustic environments, and temporal relationships. We present ThinkSound, a novel framework that leverages Chain-of-Thought (CoT) reasoning to enable stepwise, interactive audio generation and editing for videos. Our approach decomposes the process into three complementary stages: foundational foley generation that creates semantically coherent soundscapes, interactive object-centric refinement through precise user interactions, and targeted editing guided by natural language instructions. At each stage, a multimodal large language model generates contextually aligned CoT reasoning that guides a unified audio foundation model. Furthermore, we introduce AudioCoT, a comprehensive dataset with structured reasoning annotations that establishes connections between visual content, textual descriptions, and sound synthesis. Experiments demonstrate that ThinkSound achieves state-of-the-art performance in video-to-audio generation across both audio metrics and CoT metrics and excels in out-of-distribution Movie Gen Audio benchmark.
+
+---
+
+## 📰 News
+- **2025.07** &nbsp; 🔥Online demo on [Hugging Face Spaces](https://huggingface.co/spaces/FunAudioLLM/ThinkSound) and [ModelScope](https://modelscope.cn/studios/iic/ThinkSound) for interactive experience!
+- **2025.07** &nbsp; 🔥Released inference scripts and web interface; 
+- **2025.06** &nbsp; 🔥[ThinkSound paper](https://arxiv.org/pdf/2506.21448) released on arXiv!
+- **2025.06** &nbsp; 🔥[Online Demo](http://thinksound-project.github.io/) is live - try it now!
+
+---
+
+## 🚀 Features
+
+-   **Any2Audio**: Generate audio from arbitrary modalities — video, text, audio, or their combinations.
+-   **Video-to-Audio SOTA**: Achieves state-of-the-art results on multiple V2A benchmarks.
+-   **CoT-Driven Reasoning**: Chain-of-Thought reasoning for compositional and controllable audio generation via MLLMs.
+-   **Interactive Object-centric Editing**: Refine or edit specific sound events by clicking on visual objects or using text instructions.
+-   **Unified Framework**: One foundation model supports generation, editing, and interactive workflow.
+
+---
+
+## ✨ Method Overview
+
+ThinkSound decomposes audio generation and editing into three interactive stages, all guided by MLLM-based Chain-of-Thought (CoT) reasoning:
+
+1.  **Foley Generation:** Generate foundational, semantically and temporally aligned soundscapes from video.
+2.  **Object-Centric Refinement:** Refine or add sounds for user-specified objects via clicks or regions in the video.
+3.  **Targeted Audio Editing:** Modify generated audio using high-level natural language instructions.
+
+![ThinkSound Overview](https://github.com/liuhuadai/ThinkSound/blob/main/assets/figs/fig3_model.png)
+
+---
+
+## ⚡ Quick Start
+
+**Environment Preparation:**
+```bash
+git clone https://github.com/liuhuadai/ThinkSound.git
+cd ThinkSound
+pip install -r requirements.txt
+conda install -y -c conda-forge 'ffmpeg<7'
+# Download pretrained weights https://huggingface.co/liuhuadai/ThinkSound to Directory ckpts/
+# model weights can be also downloaded from https://www.modelscope.cn/models/iic/ThinkSound
+git lfs install
+git clone https://huggingface.co/liuhuadai/ThinkSound
+```
+
+**Make it executable**
+```bash
+chmod +x scripts/demo.sh
+```
+
+**Run the script**
+```bash
+./scripts/demo.sh <video_path> <title> <CoT description>
+```
+
+### Web Interface Usage
+
+For an interactive experience, launch the Gradio web interface:
+
+```bash
+python app.py
+```
+
+---
+
+## 📝 TODO
+
+-   ☐ Release training scripts for ThinkSound models
+-   ☐ Open-source AudioCoT dataset and automated pipeline
+-   ☐ Provide detailed documentation and API reference
+-   ☐ Add support for additional modalities and downstream tasks
+
+---
+
+## 📄 License
+
+This project is released under the [Apache 2.0 License](https://github.com/liuhuadai/ThinkSound/blob/main/LICENSE).
+
+> **Note:**  
+> The code, models, and dataset are **for research and educational purposes only**.  
+> **Commercial use is NOT permitted.**
+>
+> For commercial licensing, please contact the authors.
+
+---
+
+## 📖 Citation
+
+If you find ThinkSound useful in your research or work, please cite our paper:
 
 ```bibtex
 @misc{liu2025thinksoundchainofthoughtreasoningmultimodal,
-    title={ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing}, 
-    author={Huadai Liu and Jialei Wang and Kaicheng Luo and Wen Wang and Qian Chen and Zhou Zhao and Wei Xue},
-    year={2025},
-    eprint={2506.21448},
-    archivePrefix={arXiv},
-    primaryClass={eess.AS},
-    url={https://arxiv.org/abs/2506.21448},   
+      title={ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing}, 
+      author={Huadai Liu and Jialei Wang and Kaicheng Luo and Wen Wang and Qian Chen and Zhou Zhao and Wei Xue},
+      year={2025},
+      eprint={2506.21448},
+      archivePrefix={arXiv},
+      primaryClass={eess.AS},
+      url={https://arxiv.org/abs/2506.21448}, 
 }
-```
+```
+
+---
+
+## 📬 Contact
+
+✨ Feel free to [open an issue](https://github.com/liuhuadai/ThinkSound/issues) or contact us via email ([[email protected]](mailto:[email protected])) if you have any questions or suggestions!