Views
No views yet
pip install transformers torch pandas scikit-learn1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4model = AutoModelForSequenceClassification.from_pretrained("ez6047/CIREE-fact-detector")
5tokenizer = AutoTokenizer.from_pretrained("ez6047/CIREE-fact-detector")
6
7inputs = tokenizer("The Amazon river flows through Brazil.", return_tensors="pt")
8with torch.no_grad():
9 logits = model(**inputs).logits
10prob = torch.softmax(logits, dim=-1)[0, 1].item()
11print(f"Fact probability: {prob:.3f}") # threshold: 0.635run_inference.py in this repo.