import numpy as np
if __name__ == '__main__':
guid = 0
tokens = np.array(['CLS', 'i', 'want', 'to', 'fly', 'from', 'baltimore', 'to', 'dallas', 'round', 'trip', 'SEP'])
input_ids = [101, 1045, 2215, 2000, 4875, 2013, 6222, 2000, 5759, 2461, 4440, 102, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1