Fix to 'Model' object has no attribute '_container_nodes' error when using tf.keras.utils.plot_model().