Text generation LSTM in TensorFlow