huBert-fine-tuned-hungarian-squadv1

MODEL DESCRIPTION

huBERT base model (cased) fine-tuned on SQuAD v1

• huBert model + Tokenizer: https://huggingface.co/SZTAKI-HLT/hubert-base-cc
• Hungarian SQUAD v1 dataset: Machine Translated SQuAD dataset (Google Translate API)
• This is a demo model. Date of publication: 2022.03.27.

Model in action

• Fast usage with pipelines:

from transformers import pipeline
qa_pipeline = pipeline(
    "question-answering",
    model="mcsabai/huBert-fine-tuned-hungarian-squadv1",
    tokenizer="mcsabai/huBert-fine-tuned-hungarian-squadv1"
)
predictions = qa_pipeline({
    'context': "Anita vagyok és Budapesten élek már több mint 4 éve.",
    'question': "Hol lakik Anita?"
})
print(predictions)
# output:
# {'score': 0.9892364144325256, 'start': 16, 'end': 26, 'answer': 'Budapesten'}