code: def model_fn(model_dir):
      # Load model from HuggingFace Hub
      tokenizer = AutoTokenizer.from_pretrained(model_dir)
      model = AutoModel.from_pretrained(model_dir)
      return model, tokenizer
  
imports: from transformers import AutoTokenizer, AutoModel
  import torch
  import torch.nn.functional
label_desc: Use image node_id 2 for calling a global function with example usage: model_fn(model_dir) and returns: model,  tokenizer