safechat-gpt2
NoNameForMe
Clasificación de texto
Modelo de clasificación de texto basado en GPT-2, diseñado para SafeChat desarrollado por NoNameForMe. Este modelo es compatible con AutoTrain y Endpoints de Inferencia.
Como usar
Cargar el modelo utilizando AutoModelForSequenceClassification y el tokenizador AutoTokenizer.
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name = 'NoNameForMe/safechat-gpt2'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
def classify_text(text):
inputs = tokenizer(text, return_tensors='pt')
outputs = model(**inputs)
predictions = outputs.logits.argmax(-1)
return predictions
text = 'I like you. I love you'
classification = classify_text(text)
print('Clasificación:', classification)
Funcionalidades
- Clasificación de texto
- Compatible con PyTorch
- Compatible con Transformers
- Basado en gpt2
- Compatible con AutoTrain
- Compatible con Endpoints de Inferencia
Casos de uso
- Clasificación de sentimientos en mensajes de chat
- Detección de contenido ofensivo o inapropiado en tiempo real
- Filtrado de contenido en aplicaciones de chat
- Monitoreo de redes sociales