Update README.md

README.md

@@ -32,15 +32,12 @@ This is a fine-tuned deit_small_patch16_224 model trained on CUB-200-2011 from t
   - L1: 0.01
   - Orthogonal: 0.001
   - Coherence: 0.003
-- Training set size: 5994
-- Push set size: N/A
-- Test set size: 5794
 - Batch size: 128
 
 ## Dataset Description
 
 Fine-grained bird species classification dataset with 200 different bird species
-Dataset link: https://www.kaggle.com/datasets/hereisburak/pins-face-recognition
+Dataset link: https://www.vision.caltech.edu/datasets/cub_200_2011/
 
 ## Usage
 
@@ -63,4 +60,7 @@ predicted_label = outputs.logits.argmax(-1).item()
 
 ## Additional Information
 
-For more details about the implementation and training process, please visit the [GitHub repository](https://github.com/ayushnangia/ProtoViT).
+Github repo by authors of the paper ![GitHub repository][https://github.com/Henrymachiyu/ProtoViT]
+
+For more details about the implementation and training process, please visit the my fork of ProtoVit ![GitHub repository][https://github.com/ayushnangia/ProtoViT].
+