TFP Uncertainty Bands

I built a BNN in TensorFlow and I am trying to make a prediction just using the green dots(see image below), however, my uncertainty bounds do not grow/expand where there is no data. I have attached the code I used to define and build my BNN and I have tried to debug why my uncertainty bounds are not expanding and narrowing at the correct points. Any help would be appreciated.