Update README.md

README.md

@@ -11,15 +11,17 @@ metrics:
 model-index:
 - name: vit-real-fake-cls
   results: []
+datasets:
+- date3k2/raw_real_fake_images
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/date3k2/real-fake-classification/runs/3wxs9xk6)
-# vit-real-fake-cls
+# ViT Real Fake Image Classification
 
-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 [Real & Fake Images](https://huggingface.co/datasets/date3k2/raw_real_fake_images) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.0398
 - Accuracy: 0.9866
@@ -27,20 +29,6 @@ It achieves the following results on the evaluation set:
 - Recall: 0.9854
 - Precision: 0.9902
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
-
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -74,4 +62,4 @@ The following hyperparameters were used during training:
 - Transformers 4.41.0
 - Pytorch 2.3.0+cu121
 - Datasets 2.19.1
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1