utils/postprocess.py

import random

def text_mapping(model_output,text_label):
    text_list = text_label[model_output]
    text,sent = random.sample(text_list,1)[0]
    return {'label' : model_output,
            'text' : text,
            'sentiment' : sent}