Creating a TensorFlow model