def get_predict_label(pos_prob):
    """
    Convert the prediction probabilities to label
    """
    # threshold should be (1, 0.5)
    neu_threshold = 0.5
    if pos_prob >= neu_threshold:
        label, key = 1, "porn"
    else:
        label, key = 0, "not_porn"
    return label, key