christopher commited on
Commit
97e0c69
·
1 Parent(s): e8db2ab

added call in NLPmodel

Browse files
Files changed (1) hide show
  1. 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)