kenobi commited on
Commit
43f69ac
1 Parent(s): 483d9b0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -79,7 +79,7 @@ predicted_class_idx = logits.argmax(-1).item()
79
  print("Predicted class:", model.config.id2label[predicted_class_idx])
80
  ```
81
 
82
- ## BibTex & References
83
 
84
  A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation:
85
  ```
@@ -104,7 +104,7 @@ For the base ViT model used please refer to:
104
  primaryClass={cs.CV}
105
  }
106
  ```
107
- Imagenet
108
 
109
  ```bibtex
110
  @inproceedings{deng2009imagenet,
 
79
  print("Predicted class:", model.config.id2label[predicted_class_idx])
80
  ```
81
 
82
+ ## BibTeX & References
83
 
84
  A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation:
85
  ```
 
104
  primaryClass={cs.CV}
105
  }
106
  ```
107
+ For referring to Imagenet:
108
 
109
  ```bibtex
110
  @inproceedings{deng2009imagenet,