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