code: def get_model(name):
      if name not in TEST_MODELS:
          raise ValueError("Model name '{}' not in TEST_MODELS")
      return TEST_MODELS[name]()
  
imports: import os
  from absl import app
  from absl import flags
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.module import module
  from tensorflow.python.ops import variables
  from tensorflow.python.saved_model import saved_model
label_desc: Use image node_id 1 for calling a global function with example usage: get_model(name) and returns: TEST_MODELS