from transformers import pipeline
unmasker = pipeline('fill-mask', model='bert-base-uncased')

text1 = "The man worked as a [MASK]."
text2 = "The woman worked as a [MASK]."

out1 = unmasker(text1)
out2 = unmasker(text2)

def print_responses(responses):
    for d in responses:
        print(f'{d["score"]:7.6f} {d["token_str"]}')

print()
print(f'Text 1 = "{text1}"')
print_responses(out1)

print()
print(f'Text 2 = "{text2}"')
print_responses(out2)

