Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "VladimirRH/toxicity-classifier"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7model.eval()
8
9text = "Спасибо за обратную связь, мы обязательно учтём ваши пожелания."
10
11inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=32)
12
13with torch.no_grad():
14 logits = model(**inputs).logits
15 probs = torch.softmax(logits, dim=1)
16
17prediction = torch.argmax(probs, dim=1).item()
18toxicity_prob = probs[0][1].item()
19
20print(f"Класс: {prediction}")
21print(f"Вероятность токсичности: {toxicity_prob:.4f}")