fixing port bug in cifar_predicition