OnlyBiggg commited on
Commit
f7b21c2
·
1 Parent(s): 9eb61a1
Files changed (1) hide show
  1. app/ner/services/ner.py +1 -1
app/ner/services/ner.py CHANGED
@@ -22,7 +22,7 @@ class NER:
22
  async def predict(self, text: str, entity_tag: str = None):
23
  if self.pipeline is None:
24
  raise ValueError("Model not loaded. Please call load_model() first.")
25
- pred = await self.pipeline(text)
26
  if entity_tag:
27
  return self.extract_entities(pred, entity_tag)
28
  return pred
 
22
  async def predict(self, text: str, entity_tag: str = None):
23
  if self.pipeline is None:
24
  raise ValueError("Model not loaded. Please call load_model() first.")
25
+ pred = self.pipeline(text)
26
  if entity_tag:
27
  return self.extract_entities(pred, entity_tag)
28
  return pred