Views
No views yet
1from transformers import RobertaForSequenceClassification, RobertaTokenizer
2import torch
3
4tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
5model = RobertaForSequenceClassification.from_pretrained("oeg/RoBERTa-Repository-Proposal")
6
7sentence = "Your input sentence here."
8inputs = tokenizer(sentence, return_tensors="pt")
9outputs = model(**inputs)
10probabilities = torch.nn.functional.softmax(outputs.logits, dim=1)