Update embedding layer sizes
Update embedding layer sizes

The one_hot function from keras generates numbers in the range of [1,
vocab_size] (in our case), and thus should actually accept vectors of
size vocab_size+1.

googleinterns/amaranth,googleinterns/amaranth