Update README.md

README.md

@@ -4,9 +4,9 @@ tags:
 - vision
 ---
 
-# Vision Transformer (large-sized model) trained using DINOv2
+# ViTMatte model
 
-Vision Transformer (ViT) model trained using the DINOv2 method. It was introduced in the paper [ViTMatte: Boosting Image Matting with Pretrained Plain Vision Transformers](https://arxiv.org/abs/2305.15272) by Yao et al. and first released in [this repository](https://github.com/hustvl/ViTMatte). 
+ViTMatte model trained on Composition-1k. It was introduced in the paper [ViTMatte: Boosting Image Matting with Pretrained Plain Vision Transformers](https://arxiv.org/abs/2305.15272) by Yao et al. and first released in [this repository](https://github.com/hustvl/ViTMatte). 
 
 Disclaimer: The team releasing ViTMatte did not write a model card for this model so this model card has been written by the Hugging Face team.