Update README.md

README.md

@@ -23,7 +23,7 @@ A flax-based MonoBERT cross encoder trained on the [Baidu-ULTR](https://arxiv.or
 | **Naive Listwise**  | -              | 1.9738 | 4.1609 | 5.6861 | 8.5432 | 0.4091  | 0.6436 |
 
 ## Usage
-Here is an example with a mock input batch for how to download and call the model:
+Here is an example of downloading the model and calling it for inference on a mock batch of input data. For more details on how to use the model on the Baidu-ULTR dataset, take a look at our [training](https://github.com/philipphager/baidu-bert-model/blob/main/main.py) and [evaluation scripts](https://github.com/philipphager/baidu-bert-model/blob/main/eval.py) in our code repository.
 
 ```Python
 import jax.numpy as jnp
@@ -67,8 +67,6 @@ outputs = model(batch, train=False)
 print(outputs)
 ```
 
-For more details on how to use the model with real data from Baidu-ULTR, take a look at the [evaluation script of our model repository](https://github.com/philipphager/baidu-bert-model/blob/main/eval.py).
-
 ## Reference
 ```
 @inproceedings{Hager2024BaiduULTR,