code: def get_model_cls(model_name: str) -> Callable[..., Any]:
      """Returns model class given its name."""
      if model_name == "vid2seq":
          return models.DenseVideoCaptioningModel
      raise ValueError(f"Unrecognized model: {model_name}.")
  
file_path: scenic/scenic/projects/vid2seq/main.py
imports: import os
  from typing import Any, Callable
  from absl import flags
  from clu import metric_writers
  import jax
  import jax.numpy
  import ml_collections
  from scenic import app
  from scenic.projects.vid2seq import models
  from scenic.projects.vid2seq import trainer
  from scenic.projects.vid2seq.datasets.dense_video_captioning_tfrecord_dataset import get_datasets
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_cls(model_name) and returns: models