Use python to make the prediction.