Update mcp/nlp.py
Browse files- mcp/nlp.py +3 -3
mcp/nlp.py
CHANGED
@@ -7,13 +7,13 @@ def load_model():
|
|
7 |
return spacy.load("en_core_web_sm")
|
8 |
except OSError:
|
9 |
raise RuntimeError(
|
10 |
-
"spaCy model 'en_core_web_sm' not
|
11 |
-
"
|
12 |
)
|
13 |
|
14 |
nlp = load_model()
|
15 |
|
16 |
def extract_keywords(text: str):
|
17 |
-
"""Extract
|
18 |
doc = nlp(text)
|
19 |
return list({ent.text for ent in doc.ents if len(ent.text.strip()) > 2})
|
|
|
7 |
return spacy.load("en_core_web_sm")
|
8 |
except OSError:
|
9 |
raise RuntimeError(
|
10 |
+
"spaCy model 'en_core_web_sm' is not installed. "
|
11 |
+
"Ensure Dockerfile downloads it at build time."
|
12 |
)
|
13 |
|
14 |
nlp = load_model()
|
15 |
|
16 |
def extract_keywords(text: str):
|
17 |
+
"""Extract unique Named Entities longer than 2 characters."""
|
18 |
doc = nlp(text)
|
19 |
return list({ent.text for ent in doc.ents if len(ent.text.strip()) > 2})
|