Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("louistzx/kaypohaunty")
5model = AutoModelForSequenceClassification.from_pretrained("louistzx/kaypohaunty")
6
7inputs = tokenizer("Your text here", return_tensors="pt", truncation=True, max_length=256)
8with torch.no_grad():
9 outputs = model(**inputs)
10 predictions = torch.sigmoid(outputs.logits)