Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3model = AutoModelForSequenceClassification.from_pretrained("ScottBiggs2/ThradBertConversationClassifier")
4tokenizer = AutoTokenizer.from_pretrained("ScottBiggs2/ThradBertConversationClassifier")
5
6# Example inference
7text = "Your conversation text here"
8inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
9outputs = model(**inputs)
10predictions = outputs.logits.softmax(dim=-1)