Add initial version of ROBERTA classifier, add ability for multi pi charts

This commit is contained in:
William Jeynes
2026-03-11 22:02:31 +00:00
parent ef6330ec07
commit f09e36e740
6 changed files with 299 additions and 33 deletions
@@ -0,0 +1,25 @@
from transformers import RobertaTokenizer, RobertaForSequenceClassification
import torch
MODEL_PATH = "./roberta_classifier"
tokenizer = RobertaTokenizer.from_pretrained(MODEL_PATH)
model = RobertaForSequenceClassification.from_pretrained(MODEL_PATH)
text2 = "High-profile political downplaying of COVID-19 (examples: President Trump saying 'it will go away' in MarchAugust 2020)"
text = "Multiple mirrored reuploads (20202023) put the clip on other channels with titles implying it was a genuine 1970s public information film."
inputs = tokenizer(
text,
return_tensors="pt",
truncation=True,
padding=True
)
model.eval()
with torch.no_grad():
logits = model(**inputs).logits
probs = torch.softmax(logits, dim=1)
print(probs)