Views
No views yet
1from transformers import pipeline
2qa_pipeline = pipeline(
3 "question-answering",
4 model="mcsabai/huBert-fine-tuned-hungarian-squadv1",
5 tokenizer="mcsabai/huBert-fine-tuned-hungarian-squadv1"
6)
7predictions = qa_pipeline({
8 'context': "Anita vagyok és Budapesten élek már több mint 4 éve.",
9 'question': "Hol lakik Anita?"
10})
11print(predictions)
12# output:
13# {'score': 0.9892364144325256, 'start': 16, 'end': 26, 'answer': 'Budapesten'}