def predict_fn()

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

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

def predict_fn(request_body, model_assets):
    nlp = model_assets['nlp']
    text = request_body["text"]
    spacy_document = nlp(text)
    entities = extract_entities(spacy_document)
    noun_chunks = extract_noun_chunks(spacy_document)
    return {
        "entities": entities,
        "noun_chunks": noun_chunks
    }