Update README.md

README.md

@@ -10,31 +10,11 @@ model-index:
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
 probably proofread and complete it, then remove this comment. -->
 
-# darshanz/occupaion-prediction
+# darshanz/occupation-prediction
 
-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.
-It achieves the following results on the evaluation set:
-- Train Loss: 0.0378
-- Train Accuracy: 0.9953
-- Train Top-3-accuracy: 1.0
-- Validation Loss: 0.4967
-- Validation Accuracy: 0.8433
-- Validation Top-3-accuracy: 0.9800
-- Epoch: 6
+This model is ViT base patch16. Which is pretrained on imagenet dataset, then trained on our custom dataset which is based on occupation prediction. This dataset contains facial images of Indian people which are labeled by occupation. This model predicts the occupation of a person from the facial image of a person. This model gives an accuracy of 84.43%.
 
-## Model description
 
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
 
 ### Training hyperparameters