Issue on page /bayesian-code.html

[Ericyang978]

When running the colab notebook, I ran into an issue with the cell that begins the model initialization and computes kl_scaling and KL_PREF. The following is the error:

"ValueError: Only instances of keras.Layer can be added to a Sequential model. Received: <tensorflow_probability.python.layers.conv_variational.Conv2DFlipout object at 0x7f00dd10b640> (of type <class 'tensorflow_probability.python.layers.conv_variational.Conv2DFlipout'>)"