fix missing import os statement in cifar_prediction