Views
No views yet
1device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
2tokenizer = AutoTokenizer.from_pretrained("Lycoris53/roberta-base-japanese-wrime-finetuned")
3model = RobertaForSequenceClassification.from_pretrained("Lycoris53/roberta-base-japanese-wrime-finetuned", num_labels=8, problem_type='regression')
4model.to(device)
5
6text_to_predict = "おはようございます。今日も一緒に頑張りましょう!"
7inputs = tokenizer(text_to_predict, return_tensors="pt", truncation=True, max_length=512)
8inputs.to(device)
9
10with torch.no_grad(): # Disable gradient calculations for inference
11 outputs = model(**inputs)
12
13logits = F.softmax(outputs.logits.cpu()[0]).tolist()
14...