Downloads · 30 days
17
0% of all-time downloads
mcsabai/huBert-fine-tuned-hungarian-squadv2
huBert-fine-tuned-hungarian-squadv2 is a question answering model from mcsabai. Use it when the input is a question plus a passage. It is set up for transformers.
huBERT base model (cased) fine-tuned on SQuADv2 (NEW!)
Downloads · 30 days
17
0% of all-time downloads
All-time downloads
7.8K
Public
Repo size
1.3 GB
Likes
5
Public
Click a slice to open those files.
.h5440 MB · 50%
From the Hugging Face model README
huBERT base model (cased) fine-tuned on SQuADv2 (NEW!)
from transformers import pipeline
qa_pipeline = pipeline(
"question-answering",
model="mcsabai/huBert-fine-tuned-hungarian-squadv2",
tokenizer="mcsabai/huBert-fine-tuned-hungarian-squadv2",
topk = 1,
handle_impossible_answer = True
)
predictions = qa_pipeline({
'context': "Máté vagyok és Budapesten élek már több mint 4 éve.",
'question': "Hol lakik Máté?"
})
print(predictions)
# output:
# {'score': 0.9892364144325256, 'start': 16, 'end': 26, 'answer': 'Budapesten'}
Two important parameter: