Creating and Saving the TF model