DatafoundryAI
/

roberta-rerank

sentence-transformers

text-classification

text-embeddings-inference

Model card Files Files and versions

Update README.md

#1

by sabre-code - opened 6 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ Pre-trained models can be used like this:
 ```python
 from sentence_transformers import CrossEncoder
-model = CrossEncoder('cross-encoder/stsb-roberta-base')
 scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
 ```

 ```python
 from sentence_transformers import CrossEncoder
+model = CrossEncoder('DatafoundryAI/roberta-rerank')
 scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
 ```