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