How to use Bayesian Inference for predictions in Python