How would you re-write the code with Tensorflow 2 where learn.preprocessing.VocabularyProcessor and tf.contrib are deprecated