I was wondering if there are ways to create annotation file and add that to this model and dataset to have extra key or entity extraction?