Update predict.py