Accuracy and Loss Plots for Tensorflow Model 2.0

Question

I am following this tutorial https://www.youtube.com/watch?v=amURyS6CAaY exactly on training a TensorFlow model with a custom dataset. I tried adding a Tensorboard however it does not show the accuracy and loss by epoch. The issue with the below code is that it is only plotting 1 point instead of the loss over each epoch. How can this be fixed?

Here is my attempted code which is added to https://colab.research.google.com/drive/1QCU_dCR0ozI8j6X2btEDCsaUk5p_b1uw?usp=sharing:

logger = 'TensorBoard' #@param ['TensorBoard']

if logger == 'TensorBoard':
  %load_ext tensorboard
  %tensorboard --logdir '/content/gdrive/MyDrive/Colab_Notebooks/customTF2/training'

Current Result:

Expected Result: source: TensorBoard - Plot training and validation losses on the same graph?