distill-xlm-mrc / README.md
anhtunguyen98's picture
add model
a3c5573
|
raw
history blame
799 Bytes
metadata
language:
  - vi
  - vn
  - en
tags:
  - question-answering
  - pytorch
datasets:
  - squad
pipeline_tag: question-answering
metrics:
  - squad
from transformers import pipeline
model_checkpoint = "aicryptogroup/distill-xlm-mrc"
nlp = pipeline('question-answering', model=model_checkpoint,
                   tokenizer=model_checkpoint)
QA_input = {
  'question': "what is the capital of Vietnam",
  'context': "Keeping an ageless charm through centuries, Hanoi - the capital of Vietnam is famous not only for the Old Quarter with narrow and crowded streets but also for the nostalgic feeling that it brings. While Saigon is a young and modern city, the ancient Hanoi is still a true beholder of history."
}
res = nlp(QA_input)
print('pipeline: {}'.format(res))