Tag ai model implementation