Update README.md

README.md

@@ -79,7 +79,7 @@ predicted_class_idx = logits.argmax(-1).item()
 print("Predicted class:", model.config.id2label[predicted_class_idx])
 ```
 
-## References
+## BibTex & References
 
 A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation: 
 ```
@@ -103,5 +103,15 @@ For the base ViT model used please refer to:
       archivePrefix={arXiv},
       primaryClass={cs.CV}
 }
-```
 
+Imagenet
+
+```
+@inproceedings{deng2009imagenet,
+  title={Imagenet: A large-scale hierarchical image database},
+  author={Deng, Jia and Dong, Wei and Socher, Richard and Li, Li-Jia and Li, Kai and Fei-Fei, Li},
+  booktitle={2009 IEEE conference on computer vision and pattern recognition},
+  pages={248--255},
+  year={2009},
+  organization={Ieee}
+}