Multi-Output Model with TensorFlow Keras Functional API