code: def get_model_cls(model_name):
      """ "Selects Vivit model type."""
      if model_name == "vivit_classification":
          return model.ViViTModelWithObjects
      else:
          raise ValueError("Unrecognized model: {}".format(model_name))
  
file_path: scenic/scenic/projects/objectvivit/main.py
imports: from absl import flags
  from clu import metric_writers
  import jax
  import jax.numpy
  import ml_collections
  from scenic import app
  from scenic.projects.objectvivit import model
  from scenic.projects.objectvivit import trainer
  import scenic.projects.objectvivit.datasets
  from scenic.train_lib import train_utils
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_cls(model_name) and returns: model