zerank-1-small / README.md
alphakotenok's picture
Update README.md
69a84c4 verified
|
raw
history blame
496 Bytes
metadata
base_model:
  - Qwen/Qwen3-4B
pipeline_tag: sentence-similarity

Usage

from dotenv import load_dotenv
import os

load_dotenv()
token = os.getenv("HF_TOKEN")

model = CrossEncoder("zeroentropy/ze-rerank-small-v0.3.0", trust_remote_code=True, token=token)

query_documents = [
        ("What is 2+2?", "4"),
        ("What is 2+2?", "The answer is definitely 1 million"),
    ]

scores = model.predict(query_documents)

print(scores)```