class_docstr: A DoFn that loads the model and predicts.
code: def __init__(
      self,
      eval_shared_models: Dict[str, types.EvalSharedModel],
      eval_config,
  ):
      super().__init__(
          {k: v.model_loader for k, v in eval_shared_models.items()}
      )
      self._eval_config = eval_config
  
imports: import collections
  import copy
  from typing import Any, Dict, List, Optional, Sequence
  import apache_beam
  import numpy
  import pyarrow
  from tensorflow_model_analysis import constants
  from tensorflow_model_analysis.api import types
  from tensorflow_model_analysis.eval_saved_model import constants
  from tensorflow_model_analysis.extractors import extractor
  from tensorflow_model_analysis.extractors import legacy_feature_extractor
  from tensorflow_model_analysis.proto import config_pb2
  from tensorflow_model_analysis.utils import model_util
label_desc: Use image node_id 1 to create a new _TFMAPredictionDoFn object from inherited base classes: model_util with example: obj = _TFMAPredictionDoFn(eval_shared_models, eval_config)