bert-tiny-finetuned-squadv2

Uso rápido con pipelines:

from transformers import pipeline
qa_pipeline = pipeline(
"question-answering",
model="mrm8488/bert-tiny-finetuned-squadv2",
tokenizer="mrm8488/bert-tiny-finetuned-squadv2")

qa_pipeline({
'context': "Manuel Romero ha estado trabajando arduamente en el repositorio hugginface/transformers últimamente",
'question': "¿Quién ha estado trabajando arduamente para hugginface/transformers últimamente?"
})

# Salida:

{
"answer": "Manuel Romero",
"end": 13,
"score": 0.05684709993458714,
"start": 0
}

¡Sí! Eso fue fácil 🎉 Probemos con otro ejemplo

qa_pipeline({
'context': "Manuel Romero ha estado trabajando arduamente en el repositorio hugginface/transformers últimamente",
'question': "¿Para qué empresa ha trabajado Manuel Romero?"
})

# Salida:

{
"answer": "hugginface/transformers",
"end": 79,
"score": 0.11613431826808274,
"start": 56
}