recobo/chemical-bert-uncased-squad2
recobo
Pregunta y respuesta
Modelo de respuesta a preguntas utilizando Transformers y PyTorch. Ideal para conversiones de modelos entre pytorch y transformers para proporcionar libertad al usuario y permitir fácilmente cambiar entre marcos.
Como usar
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
model_name = "recobo/chemical-bert-uncased-squad2"
# Obtener predicciones
dlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
QA_input = {
'question': '¿Por qué es importante la conversión de modelos?',
'context': 'La opción de convertir modelos entre pytorch y transformers da libertad al usuario y permite que las personas cambien fácilmente entre marcos.'
}
res = nlp(QA_input)
# Cargar modelo y tokenizer
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
Funcionalidades
- Respuesta a preguntas
- Basado en Transformers
- Utiliza PyTorch
- Disponible para puntos finales de inferencia
Casos de uso
- Conversión de modelos entre PyTorch y Transformers
- Implementación en puntos finales de inferencia dedicados