from transformers import pipeline
question_answerer = pipeline("question-answering", model='bert-large-uncased-whole-word-masking-finetuned-squad')

context = r"""
cube1 x value is 201.  cube1 y value is 1.65.
cube1's orientation is upright.
cube1 visibility is true.
marker2 orientation is sideways.
marker2 visibiity is true.
marker3 visibility is false.
"""

print(context)

while (True):
    query = input('Query? ')
    result = question_answerer(question=query, context=context)
    score = round(result['score'], 4)
    if score > 0.3:
        print(f"Answer: '{result['answer']}', score: {round(result['score'], 4)}, start: {result['start']}, end: {result['end']}")
    else:
        print(f"I don't know.  score={score}")
    print()