The get_model function provides a way to quickly download a prebuilt model for easy testing of leap-ie.

preprocessing_fn, model, class_list = get_model(model_name)


  • model_name (str): Name of a Torchvision or Hugging Face model.

    • Required: Yes

    • Default: None

    • Examples:

      • Torchvision:get_model('torchvision.resnet18')

      • Hugging Face: get_model('nateraw/vit-age-classifier')


  • preprocessing_fn

    • The preprocessing function used on inputs for inference.

  • model

    • The model.

  • class_list

    • List of class names corresponding to the model's output classes.