Views
No views yet
1>>> import torch
2>>> from transformers import AutoModel, AutoTokenizer
3>>>
4>>> model_id = "rmihaylov/roberta-base-sentiment-bg"
5>>> model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
6>>> tokenizer = AutoTokenizer.from_pretrained(model_id)
7>>>
8>>> inputs = tokenizer.batch_encode_plus(['Това е умно.', 'Това е тъпо.'], return_tensors='pt')
9>>> outputs = model(**inputs)
10>>> torch.softmax(outputs, dim=1).tolist()
11
12[[0.0004746630438603461, 0.9995253086090088],
13 [0.9986956715583801, 0.0013043134240433574]]