Update README.md

README.md

@@ -2,25 +2,9 @@
 library_name: keras
 ---
 
-## Model description
+This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. ImageNet-1k dataset was used for training purposes. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-jax-weights-vitb16.ipynb) to know how the porting was done.
 
-More information needed
+## References
 
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
- ## Training Metrics
-Model history needed
- ## Model Plot
-
-<details>
-<summary>View Model Plot</summary>
-
-![Model Image](./model.png)
-
-</details>
+[1] An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale: https://arxiv.org/abs/2010.11929
+[2] How to train your ViT? Data, Augmentation, and Regularization in Vision Transformers: https://arxiv.org/abs/2106.10270