Update predictor.py