def model_fn()

in right_size_your_sagemaker_endpoints/inference.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def model_fn(model_dir):
    model_path = f'{model_dir}/infer-pytorch-ic-resnet50.pt'
    model = torch.load(model_path)
    model.eval()
    return model