How to include callbacks for plotting Training and Validation Loss Graphy?

Hello all
I am trying to use this repo., but I dont know how I can edit the codes so as (1) To plot the training and validation loss graphy

(2) To make prediction to get the predictions probability


Any help please

Thanks