| from transformers import AutoModel | |
| if __name__ == "__main__": | |
| tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased") | |
| model = AutoModel.from_pretrained("sabersol/bert-base-uncased-emotion") | |
| pred = transformers.pipeline("text-classification", model=model, tokenizer=tokenizer, device=0, return_all_scores=True) | |
| example = "The music I am listening to is so energetic and makes me motivated." | |
| results = pred(example) | |
| print(results) | |