new_model = keras.models.load_model(
                "/content/gdrive/My Drive/ColabNotebooks/ckpt4/my_model.h5"
            )

tf.saved_model.save(
    model,
    "/content/gdrive/My Drive/Colab Notebooks/ckpt4/my_model.h5"
)