Modifed prediction.py.

You have to accept.