bongsoo commited on
Commit
fb4f7c7
1 Parent(s): fdb36fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
README.md CHANGED
@@ -18,8 +18,12 @@ language:
18
  Pre-trained models can be used like this:
19
  ```
20
  from sentence_transformers import CrossEncoder
21
- model = CrossEncoder('model_name')
22
- scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
 
 
 
 
23
  ```
24
 
25
  The model will predict scores for the pairs `('Sentence 1', 'Sentence 2')` and `('Sentence 3', 'Sentence 4')`.
 
18
  Pre-trained models can be used like this:
19
  ```
20
  from sentence_transformers import CrossEncoder
21
+ model = CrossEncoder('bongsoo/albert-small-kor-cross-encoder-v1')
22
+ scores = model.predict([('대한민국', '제주도'), ('대한민국', '서울')])
23
+ print(scores)
24
+ ```
25
+ ```
26
+ [0.3342184 0.23371117]
27
  ```
28
 
29
  The model will predict scores for the pairs `('Sentence 1', 'Sentence 2')` and `('Sentence 3', 'Sentence 4')`.