Loading annotation for captum.ai