OnlyBiggg commited on
Commit
9239ad2
·
1 Parent(s): 05b674b

fix truncat maxlength

Browse files
Files changed (2) hide show
  1. app/ner/services/ner.py +2 -0
  2. core/conf.py +2 -0
app/ner/services/ner.py CHANGED
@@ -17,6 +17,8 @@ class NER:
17
  self.pipeline = pipeline(task=settings.TASK_NAME,
18
  model=self.model,
19
  tokenizer=self.tokenizer,
 
 
20
  device=settings.DEVICE)
21
 
22
  async def predict(self, text: str, entity_tag: str = None):
 
17
  self.pipeline = pipeline(task=settings.TASK_NAME,
18
  model=self.model,
19
  tokenizer=self.tokenizer,
20
+ truncation=settings.TRUNCATE,
21
+ max_length=settings.MAX_LENGTH,
22
  device=settings.DEVICE)
23
 
24
  async def predict(self, text: str, entity_tag: str = None):
core/conf.py CHANGED
@@ -36,6 +36,8 @@ class Settings(BaseSettings):
36
  TASK_NAME: str = 'token-classification'
37
  ACCELERATOR: str = 'ort'
38
  DEVICE: str = 'cpu'
 
 
39
 
40
  # FastAPI
41
  FASTAPI_API_V1_PATH: str = '/api/v1'
 
36
  TASK_NAME: str = 'token-classification'
37
  ACCELERATOR: str = 'ort'
38
  DEVICE: str = 'cpu'
39
+ TRUNCATE: bool = True
40
+ MAX_LENGTH: int = 512
41
 
42
  # FastAPI
43
  FASTAPI_API_V1_PATH: str = '/api/v1'