cross-encoder
/

stsb-TinyBERT-L4

sentence-transformers

text-classification

Model card Files Files and versions Community

tomaarsen HF staff commited on 25 days ago

Commit

35262a5

·

verified ·

1 Parent(s): 3e39595

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -11,9 +11,10 @@ This model was trained on the [STS benchmark dataset](http://ixa2.si.ehu.eus/sts
 ## Usage and Performance
 Pre-trained models can be used like this:
-```
 from sentence_transformers import CrossEncoder
-model = CrossEncoder('model_name')
 scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
 ```

 ## Usage and Performance
 Pre-trained models can be used like this:
+```python
 from sentence_transformers import CrossEncoder
+model = CrossEncoder('cross-encoder/stsb-TinyBERT-L4')
 scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
 ```