Data format layoutlm

[dvjohnny]

i have model trained with data format layoutlm like this

text | labels | x0 | y0 | x1 | y1
Aragorn was the heir of Isildur | 1 | [10, 20, 30, 40, 50, 60] | [10, 10, 10, 10, 20, 20] | [20, 30, 40, 50, 60, 70] | [20, 20, 20, 20, 30, 40]
Frodo was the heir of Isildur | 0 | [15, 20, 30, 40, 50, 60] | [10, 10, 10, 10, 20, 20] | [20, 30, 45, 50, 60, 70] | [20, 20, 20, 20, 30, 40]

model = ClassificationModel(
    "layoutlm", pick_model_path,
    use_cuda=False
)

predictions, raw_outputs = model.predict(['test abc def']) 

// i got error

File "/mnt/data/home/tvkhuong/Data/SOFT/anaconda3/envs/classify/lib/python3.7/site-packages/simpletransformers/classification/classification_model.py", line 1708, in load_and_cache_examples example.label = self.args.labels_map[example.label] AttributeError: 'list' object has no attribute 'label'

how to define input message pls?