I train a pre trained BERT model on my data. I try to make a Json containing two list: first: a list conclude prediction of model (desire value) second: a