philipphager
/

baidu-ultr_uva-bert_twotower

Inference Endpoints

Model card Files Files and versions Community

philipphager commited on Apr 29

Commit

bbc753b

•

1 Parent(s): 12c676c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ import jax.numpy as jnp
 from src.model import PBMCrossEncoder
 model = PBMCrossEncoder.from_pretrained(
-    "philipphager/baidu-ultr_uva-bert_ips-pointwise",
 )
 # Mock batch following Baidu-ULTR with 4 documents, each with 8 tokens

 from src.model import PBMCrossEncoder
 model = PBMCrossEncoder.from_pretrained(
+    "philipphager/baidu-ultr_uva-bert_twotower",
 )
 # Mock batch following Baidu-ULTR with 4 documents, each with 8 tokens