Update README.md

README.md

@@ -8,6 +8,8 @@ metrics:
 model-index:
 - name: vit-base-patch16-224-dmae-va-U5-42B
   results: []
+datasets:
+- Augusto777/dmae-ve-U5
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-patch16-224-dmae-va-U5-42B
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on Augusto777/dmae-ve-U5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7215
 - Accuracy: 0.85
@@ -97,4 +99,4 @@ The following hyperparameters were used during training:
 - Transformers 4.38.2
 - Pytorch 2.2.1+cu121
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2