Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: motheecreator/vit-base-patch16-224-in21k-finetuned
+base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
 datasets:
@@ -8,7 +8,7 @@ datasets:
 metrics:
 - accuracy
 model-index:
-- name: vit-base-patch16-224-in21k-finetuned
+- name: Facial Expression Recognition
   results:
   - task:
       name: Image Classification
@@ -30,7 +30,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-patch16-224-in21k-finetuned
 
-This model is a fine-tuned version of [motheecreator/vit-base-patch16-224-in21k-finetuned](https://huggingface.co/motheecreator/vit-base-patch16-224-in21k-finetuned) on the image_folder dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the FER 2013 and MMI datasets.
 It achieves the following results on the evaluation set:
 - Loss: 0.4353
 - Accuracy: 0.8571
@@ -73,10 +73,10 @@ The following hyperparameters were used during training:
 | 0.5242        | 4.0   | 3192 | 0.7839   | 0.6282          |
 | 0.4321        | 5.0   | 3990 | 0.7823   | 0.6423          |
 | 0.3129        | 6.0   | 4788 | 0.7838   | 0.6533          |
-| 0.4245        | 7.0   | 5586 | 0.4382   | 0.8542          |
-| 0.3806        | 8.0   | 6384 | 0.4375   | 0.8531          |
-| 0.3112        | 9.0   | 7182 | 0.4372   | 0.8557          |
-| 0.2692        | 10.0  | 7980 | 0.4353   | 0.8571          |
+| 0.4245        | 7.0   | 5586 | 0.8542   | 0.4382          |
+| 0.3806        | 8.0   | 6384 | 0.8531   | 0.4375          |
+| 0.3112        | 9.0   | 7182 | 0.8557   | 0.4372          |
+| 0.2692        | 10.0  | 7980 | 0.8571   | 0.4353          |
 
 
 ### Framework versions