code: def load_embedding(
      src: str, embedding_type, layers
  ) -> TransformerEmbedding:
      """
      Load a pre-trained tensorflow text embedding from a directory.
      See the `:py:class:~alibi_detect.models.tensorflow.TransformerEmbedding` documentation for a
      full description of the `embedding_type` and `layers` kwargs.
  
      Parameters
      ----------
      src
          Name of or path to the model.
      embedding_type
         Type of embedding to extract. Needs to be one of pooler_output,
         last_hidden_state, hidden_state or hidden_state_cls.
      layers
          A list with int's referring to the hidden layers used to extract the embedding.
  
      Returns
      -------
      The loaded embedding.
      """
      emb = TransformerEmbedding(
          src, embedding_type=embedding_type, layers=layers
      )
      return emb
  
imports: import logging
  import os
  from importlib import import_module
  import warnings
  from functools import partial
  from pathlib import Path
  from typing import Callable, Dict, List, Optional, Tuple, Union, Type
  import dill
  import tensorflow
  from tensorflow_probability.python.distributions.distribution import Distribution
  from transformers import AutoTokenizer
  from alibi_detect.ad import AdversarialAE, ModelDistillation
  from alibi_detect.ad.adversarialae import DenseHidden
  from alibi_detect.cd import ChiSquareDrift, ClassifierDrift, KSDrift, MMDDrift, TabularDrift
  from alibi_detect.cd.tensorflow import UAE, HiddenOutput
  from alibi_detect.cd.tensorflow.preprocess import _Encoder
  from alibi_detect.models.tensorflow import PixelCNN, TransformerEmbedding
  from alibi_detect.models.tensorflow.autoencoder import AE, AEGMM, VAE, VAEGMM, DecoderLSTM, EncoderLSTM, Seq2Seq
  from alibi_detect.od import LLR, IForest, Mahalanobis, OutlierAE, OutlierAEGMM, OutlierProphet, OutlierSeq2Seq, OutlierVAE, OutlierVAEGMM, SpectralResidual
  from alibi_detect.od.llr import build_model
  from alibi_detect.utils.tensorflow.kernels import DeepKernel
  from alibi_detect.utils.frameworks import Framework
  from alibi_detect.version import __version__
  from alibi_detect.base import Detector
  from alibi_detect.saving._typing import VALID_DETECTORS
label_desc: Use image node_id 5 for calling a global function with example usage: load_embedding(src, embedding_type, layers) and returns: emb