def model_fn()

in sagemaker_notebook_instance/containers/entity_recognition/entry_point.py [0:0]

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

def model_fn(model_dir):
    spacy_model = "en_core_web_md"
    nlp = spacy.load(spacy_model)
    model_assets = {
        "nlp": nlp
    }
    return model_assets