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})
|