code: def get_model_cls(model_name: str):
      """Get the model class for the AdaTape project."""
      if model_name == "adatape" or model_name == "adatape-parity":
          return adatape_vit.AdaTapeMultiLabelClassificationModel
      else:
          raise ValueError(f"Unrecognized model: {model_name}.")
  
file_path: scenic/scenic/projects/adatape/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.adatape.adatape_vit import adatape_classify_trainer
  from scenic.projects.adatape.adatape_vit import adatape_trainer
  from scenic.projects.adatape.adatape_vit import adatape_vit
  from scenic.projects.adatape.dataset import parity_dataset
  from scenic.train_lib import train_utils
  from scenic.train_lib import trainers
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_cls(model_name) and returns: adatape_vit