Update README.md

README.md

@@ -6,7 +6,7 @@ datasets:
 metrics:
 - accuracy
 model-index:
-- name: resnet-mobilenet-beans
+- name: vit-mobilenet-beans-224
   results:
   - task:
       name: Image Classification
@@ -26,24 +26,14 @@ model-index:
 <!-- 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. -->
 
-# resnet-mobilenet-beans
+# ViT distilled to MobileNet
 
-This model is a fine-tuned version of [](https://huggingface.co/) on the beans dataset.
+This model is a distilled model, where teacher model is [merve/beans-vit-224](https://huggingface.co/merve/beans-vit-224), fine-tuned [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the beans dataset.
+Student model is randomly initialized MobileNetV2.
 It achieves the following results on the evaluation set:
 - Loss: 0.5922
 - Accuracy: 0.7266
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
 
 ## Training procedure