bge-reranker-v2-m3-multirag-finetuned

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tokenizer = AutoTokenizer.from_pretrained("pedrovo9/bge-reranker-v2-m3-multirag-finetuned")
model = AutoModelForSequenceClassification.from_pretrained("pedrovo9/bge-reranker-v2-m3-multirag-finetuned")

query = "What is cloud computing?"
document = "Cloud computing is..."

inputs = tokenizer(query, document, return_tensors='pt', truncation=True, max_length=512)
with torch.no_grad():
    score = torch.sigmoid(model(**inputs).logits[0, 0]).item()