chore(KaggleAux): updated predict.py