code: def get_model(model_name):
      r"""Automatically select model class based on model name
  
      Notes:
          model_name should be lowercase!
  
      Args:
          model_name (str): model name
  
      Returns:
          Generator: model class
      """
      if model_name.lower() in PLM_MODELS:
          model_name = "Pretrained_Models"
      elif model_name.lower() in RNN_MODELS:
          model_name = "RNN_Models"
      module_path = ".".join(["...model", model_name.lower()])
      if importlib.util.find_spec(module_path, __name__):
          model_module = importlib.import_module(module_path, __name__)
          model_class = getattr(model_module, model_name)
      else:
          raise ValueError("{} can't be found".format(model_name))
      return model_class
  
imports: import os
  import datetime
  import importlib
  from accelerate.logging import get_logger
  from typing import Union, Optional
  import shutil
  import torch
  from accelerate.utils import set_seed
  from transformers import AutoTokenizer, BertTokenizer
  from .enum_type import PLM_MODELS, RNN_MODELS
label_desc: Use image node_id 9 for calling a global function with example usage: get_model(model_name) and returns: model_class