Update README.md

README.md

@@ -3,7 +3,8 @@
 This model is further pre-trained on the Mathematics StackExchange questions and answers. It is based on Albert base v2 and uses the same tokenizer. In addition to pre-training the model was finetuned on Math Question Answer Retrieval. The sequence classification head is trained to output a relevance score if you input the question as the first segment and the answer as the second segment. You can use the relevance score to rank different answers for retrieval.
 
 ## Usage
-```
+```python
+# based on https://huggingface.co/docs/transformers/main/en/task_summary#sequence-classification
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
@@ -39,7 +40,7 @@ for i in range(len(classes)):
 ## Reference
 If you use this model, please consider referencing our paper:
 
-```
+```bibtex
 @inproceedings{reusch2021tu_dbs,
   title={TU\_DBS in the ARQMath Lab 2021, CLEF},
   author={Reusch, Anja and Thiele, Maik and Lehner, Wolfgang},