sabre-code commited on
Commit
e7e6666
·
verified ·
1 Parent(s): 835010c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ Pre-trained models can be used like this:
24
  ```python
25
  from sentence_transformers import CrossEncoder
26
 
27
- model = CrossEncoder('cross-encoder/stsb-roberta-base')
28
  scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
29
  ```
30
 
 
24
  ```python
25
  from sentence_transformers import CrossEncoder
26
 
27
+ model = CrossEncoder('DatafoundryAI/roberta-rerank')
28
  scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
29
  ```
30