Update README.md

README.md

@@ -1,6 +1,12 @@
+---
+license: apache-2.0
+base_model:
+- nvidia/audio-flamingo-2
+pipeline_tag: feature-extraction
+---
 # AFClap - Audio Flamingo CLAP
 
-Audio-Flamingo-2のCLAPモデル（epoch_16.pt）。テキストと音声の埋め込みを取得し、類似度計算が可能。
+Audio-Flamingo-2のCLAPモデル（epoch_16.pt）。
 
 ## Model Info
 
@@ -51,17 +57,3 @@ similarities = torch.tensor(audio_embed) @ torch.tensor(text_embed).t()
 print(similarities)
 ```
 
-## License
-
-MIT License (NVIDIA Corporation)
-
-## Citation
-
-```bibtex
-@article{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 Wang, Wei and Valle, Rafael and Catanzaro, Bryan},
-  journal={arXiv preprint arXiv:2402.01831},
-  year={2024}
-}
-```