Views
No views yet
1from transformers import pipeline
2huggingface_model = 'sidmangalik/selfBERTa'
3model = RobertaForSequenceClassification.from_pretrained(huggingface_model)
4tokenizer = RobertaTokenizerFast.from_pretrained(huggingface_model, max_length = 512, padding="max_length", truncation=True)
5
6texts = ["I am the coolest person I know."]
7
8inputs = tokenizer(texts, max_length=512, padding="max_length", truncation=True, return_tensors='pt')
9outputs = model(**inputs)
10logits = outputs.logits
11soft_logits = torch.softmax(logits, dim=1).tolist()
12predicted_classes = np.argmax(soft_logits, axis=1)