The code is written to predict POS one among {Possive Pronoun - PRP$ and Pronoun with verb - PRP_VRB} as the task was to fill blanks in given sentence from one of the two words - {"your" or "you're"}. But the code can be used to predict any of the POS.