Update README.md

README.md

@@ -17,7 +17,13 @@ A flax-based MonoBERT cross encoder trained on the Baidu-ULTR dataset with a **l
 
 ## Usage
 ```
+from src.model import ListwiseCrossEncoder
 
+model = ListwiseCrossEncoder.from_pretrained(
+    "philipphager/baidu-ultr_uva-bert_naive-listwise",
+)
+
+model(batch)
 ```
 
 ## Test Results on Baidu-ULTR Expert Annotations