AI integration: pick the embedding layer before the model | WANFENG