Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
4model = AutoModelForSequenceClassification.from_pretrained("mishrabp/mini-transformers")
5
6inputs = tokenizer("I love this!", return_tensors="pt")
7print(model(**inputs).logits)