Views
No views yet
1from transformers import pipeline
2qa_pipeline = pipeline(
3 "question-answering",
4 model="isel-nlp-mp05/bert-base-portuguese-cased-finetuned-squadv1-pt",
5 tokenizer="isel-nlp-mp05/bert-base-portuguese-cased-finetuned-squadv1-pt"
6)
7predictions = qa_pipeline({
8 'context': "Chamo-me Mattheo e estou a estudar em Lisboa.",
9 'question': "Onde é que o Mattheo está a estudar?"
10})
11print(predictions)
12# output:
13# {'score': 0.967152, 'start': 38, 'end': 44, 'answer': 'Lisboa'}