code: def load_model(model_type, checkpoint=None):
      if checkpoint is None:
          model, _, _ = load_checkpoint(model_type=model_type)
      else:
          model, _, _ = load_checkpoint(checkpoint=checkpoint)
      return model
  
file_path: detoxify/detoxify/detoxify.py
imports: import torch
  import transformers
label_desc: Use image node_id 3 for calling a global function with example usage: load_model(model_type, checkpoint) and returns: model