add predict.py