from transformers import pipeline
question_answerer = pipeline("question-answering", model='distilbert-base-cased-distilled-squad')

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

while (True):
    print()
    query = input('Query? ')
    result = question_answerer(question=query, context=context)
    print(f"Answer: '{result['answer']}', score: {round(result['score'], 4)}, start: {result['start']}, end: {result['end']}")
