Ajust rnn savemodel output