Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
4model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
5
6input_text = "The movie was fantastic with a gripping storyline!"
7inputs = tokenizer.encode(input_text, return_tensors="pt")
8outputs = model(inputs)
9print(outputs.logits)