detoxis-spanish-toxicity-classifier

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
import numpy as np

model = AutoModelForSequenceClassification.from_pretrained("Juanpeg1729/detoxis-spanish-toxicity-classifier")
tokenizer = AutoTokenizer.from_pretrained("Juanpeg1729/detoxis-spanish-toxicity-classifier")

thresholds = [1.0, 0.9800000000000001, 0.7299999999999999, 1.2499999999999998]

text = "Tu texto aquí"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
with torch.no_grad():
    outputs = model(**inputs)
    probs = torch.softmax(outputs.logits, dim=-1).numpy()[0]
    adjusted = probs * np.array(thresholds)
    prediction = adjusted.argmax()