create predict.py, which actually predicts things