order = sorted([str(i) for i in range(1098)])
chars = sorted(set(np.load("data_processed/dataY.npy")))

def fix_predictions(output):
    outs = [chars[int(order[i.item()])] for i in torch.topk(output, 5, dim = 1).indices[0]]
    return ["\\" + i.split("_")[1] for i in outs]