csarron/mobilebert-uncased-squad-v1

csarron
Pregunta y respuesta

MobileBERT es una versión delgada de BERT_LARGE, equipado con estructuras de cuello de botella y un equilibrio cuidadosamente diseñado entre auto-atenciones y redes de alimentación directa. Este modelo fue ajustado desde el punto de control de HuggingFace google/mobilebert-uncased en SQuAD1.1.

Como usar

from transformers import pipeline

qa_pipeline = pipeline(
    'question-answering',
    model='csarron/mobilebert-uncased-squad-v1',
    tokenizer='csarron/mobilebert-uncased-squad-v1')

predictions = qa_pipeline({
    'context': "The game was played on February 7, 2016 at Levi's Stadium in the San Francisco Bay Area at Santa Clara, California.",
    'question': "What day was the game played on?"
})

print(predictions)
# output:
# {'score': 0.7754058241844177, 'start': 23, 'end': 39, 'answer': 'February 7, 2016'}

Funcionalidades

Ajustado en SQuAD v1
Arquitectura MobileBERT
Basado en Transformers
Implementación en PyTorch
Usa Safetensors
Licencia MIT

Casos de uso

Responder preguntas basadas en contexto
Aplicaciones de ayuda virtual y chatbots
Extracción de información