Update README.md

README.md

@@ -1,9 +1,25 @@
 ---
+license: cc-by-nc-4.0
 tags:
-- model_hub_mixin
-- pytorch_model_hub_mixin
+- audio-to-video
+library_name: diffusers
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: [More Information Needed]
-- Docs: [More Information Needed]
+Models of [R-FLAV](https://arxiv.org/abs/2503.08307) trained on Landscape and AIST++ for 400k iterations.
+
+For more info, please refer to the Github repository at https://github.com/MaverickAlex/R-FLAV
+
+To download the ckpts directly in the code you can do
+```python
+from huggingface_hub import hf_hub_download
+import torch
+from models import FLAV
+
+model = FLAV.from_pretrained(args.model_ckpt)
+
+hf_hub_download(repo_id="MaverickAlex/R-FLAV-B-1-LS", filename="vocoder/config.json")
+vocoder_path = hf_hub_download(repo_id="MaverickAlex/R-FLAV-B-1-LS", filename="vocoder/vocoder.pt")
+
+vocoder_path = vocoder_path.replace("vocoder.pt", "")
+vocoder = Generator.from_pretrained(vocoder_path)
+```