christopher
commited on
Commit
·
97e0c69
1
Parent(s):
e8db2ab
added call in NLPmodel
Browse files- models/nlp.py +4 -0
models/nlp.py
CHANGED
|
@@ -5,6 +5,10 @@ class NLPModel:
|
|
| 5 |
def __init__(self):
|
| 6 |
self.nlp = spacy.load("pt_core_news_md")
|
| 7 |
nltk.download('punkt')
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
def extract_entities(self, text: str):
|
| 10 |
doc = self.nlp(text)
|
|
|
|
| 5 |
def __init__(self):
|
| 6 |
self.nlp = spacy.load("pt_core_news_md")
|
| 7 |
nltk.download('punkt')
|
| 8 |
+
|
| 9 |
+
def __call__(self, text: str):
|
| 10 |
+
"""Makes the model callable like model(text)."""
|
| 11 |
+
return self.extract_entities(text) # or another default method
|
| 12 |
|
| 13 |
def extract_entities(self, text: str):
|
| 14 |
doc = self.nlp(text)
|