Create predict.py