hi @AviPaul! Essentially it’s the same model (a CRF) which does the entity extraction in the first place, it’s just a second CRF which is integrated into the DIET architecture. You can also learn more from Tanja’s post here