spark_nlp_pretrained_pipeline_text_annotation_quickstart.py

1import sparknlp
2from sparknlp.pretrained import PretrainedPipeline
3
4# Start Spark Session with Spark NLP
5spark = sparknlp.start()
6
7# Load a pre-trained pipeline (Explain Document ML)
8pipeline = PretrainedPipeline("explain_document_ml", lang="en")
9
10# Annotate a sample text
11text = "The Mona Lisa is a 16th century oil painting created by Leonardo. It's held at the Louvre in Paris."
12result = pipeline.annotate(text)
13
14# Check the results
15print(result.keys())
16print(result['entities'])