Views
No views yet
1import torch
2from transformers import RobertaForSequenceClassification, RobertaTokenizer
3# Load the fine-tuned model
4model = RobertaForSequenceClassification.from_pretrained('Dzeniks/justification-analyst')
5
6# Load the tokenizer
7tokenizer = RobertaTokenizer.from_pretrained('Dzeniks/justification-analyst')
8
9# Tokenize the input sequence
10input_text = "This is a sample input sequence"
11input = tokenizer.encode_plus(claim, evidence, return_tensors="pt")
12# Use the model to make a prediction
13model.eval()
14with torch.no_grad():
15 prediction = model(**x)
16predictions = torch.argmax(outputs[0], dim=1).item()