Update README.md

README.md

@@ -1,63 +1,14 @@
-# PyTorch Implementation of Audio Flamingo 2
-
-**Sreyan Ghosh, Zhifeng Kong, Sonal Kumar, S Sakshi, Jaehyeon Kim, Wei Ping, Rafael Valle, Dinesh Manocha, Bryan Catanzaro**
-
-[[paper]](https://arxiv.org/abs/2503.03983) [[Demo website]](https://research.nvidia.com/labs/adlr/AF2/) [[GitHub]](https://github.com/NVIDIA/audio-flamingo)
-
-This repo contains the PyTorch implementation of [Audio Flamingo 2: An Audio-Language Model with Long-Audio Understanding and Expert Reasoning Abilities](). Audio Flamingo 2 achieves the state-of-the-art performance across over 20 benchmarks, with only a 3B parameter small language model. It is improved from our previous [Audio Flamingo](https://arxiv.org/abs/2402.01831). 
-
-- We introduce two datasets, AudioSkills for expert audio reasoning, and LongAudio for long audio understanding, to advance the field of audio understanding.
-
-- Audio Flamingo 2 has advanced audio understanding and reasoning capabilities. Especially, Audio Flamingo 2 has expert audio reasoning abilities, and can understand long audio up to 5 minutes.
-
-- Audio Flamingo 2 outperforms larger and proprietary LALMs across 20+ benchmarks, despite being smaller (3B) and trained exclusively on public datasets.
-
-## Main Results
-
-Audio Flamingo 2 outperforms prior SOTA models including GAMA, Audio Flamingo, Qwen-Audio, Qwen2-Audio, LTU, LTU-AS, SALMONN, AudioGPT, Gemini Flash v2, Gemini Pro v1.5, and GPT-4o-audio on a number of understanding and reasoning benchmarks.
-
-<div align="center">
-  <img class="img-full" src="assets/af2_radar.png" width="300">
-</div>
-
-<div align="center">
-  <img class="img-full" src="assets/af2_table2.png" width="400">
-</div>
-
-## Audio Flamingo 2 Architecture
-
-Audio Flamingo 2 uses a cross-attention architecture similar to [Audio Flamingo](https://arxiv.org/abs/2402.01831) and [Flamingo](https://arxiv.org/abs/2204.14198). Audio Flamingo 2 can take up to 5 minutes of audio inputs. 
-
-<div align="center">
-  <img class="img-full" src="assets/af2_arch.png" width="800">
-</div>
-
-## License
-
-- The checkpoints are for non-commercial use only (see NVIDIA OneWay Noncommercial License). They are also subject to the [Qwen Research license](https://huggingface.co/Qwen/Qwen2.5-3B/blob/main/LICENSE), the [Terms of Use](https://openai.com/policies/terms-of-use) of the data generated by OpenAI, and the original licenses accompanying each training dataset.
-- Notice: Audio Flamingo 2 is built with Qwen-2.5. Qwen is licensed under the Qwen RESEARCH LICENSE AGREEMENT, Copyright (c) Alibaba Cloud. All Rights Reserved.
-
-
-## Citation
-
-- Audio Flamingo
-```
-@inproceedings{kong2024audio,
-  title={Audio Flamingo: A Novel Audio Language Model with Few-Shot Learning and Dialogue Abilities},
-  author={Kong, Zhifeng and Goel, Arushi and Badlani, Rohan and Ping, Wei and Valle, Rafael and Catanzaro, Bryan},
-  booktitle={International Conference on Machine Learning},
-  pages={25125--25148},
-  year={2024},
-  organization={PMLR}
-}
-```
-
-- Audio Flamingo 2
-```
-@article{ghosh2025audio,
-  title={Audio Flamingo 2: An Audio-Language Model with Long-Audio Understanding and Expert Reasoning Abilities},
-  author={Ghosh, Sreyan and Kong, Zhifeng and Kumar, Sonal and Sakshi, S and Kim, Jaehyeon and Ping, Wei and Valle, Rafael and Manocha, Dinesh and Catanzaro, Bryan},
-  journal={arXiv preprint arXiv:2503.03983},
-  year={2025}
-}
-```
+---
+title: Audio Flamingo 2
+emoji: 🏃
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+sdk_version: 5.15.0
+app_file: app.py
+pinned: false
+license: apache-2.0
+short_description: Audio Flamingo 2 Demo
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference