Files

 main

/

reranker.py

Copy path

Blame

Blame

Latest commit

 

History

History

9 lines (6 loc) · 420 Bytes

 main

/

reranker.py

Top

File metadata and controls

• Code
• Blame

9 lines (6 loc) · 420 Bytes

Raw

1

2

3

4

5

6

7

8

9

from embed_anything import Reranker, Dtype, RerankerResult, DocumentRank

reranker = Reranker.from_pretrained("jinaai/jina-reranker-v1-turbo-en", dtype=Dtype.F16)

results: list[RerankerResult] = reranker.rerank(["What is the capital of France?"], ["France is a country in Europe.", "Paris is the capital of France."], 2)

for result in results:

documents: list[DocumentRank] = result.documents

print(documents)