change code to use predicted pos/lemma/feats