Update predictions.py