OnlyBiggg commited on
Commit
3657839
·
1 Parent(s): 309a937

fix trunca

Browse files
Files changed (1) hide show
  1. app/ner/services/ner.py +3 -3
app/ner/services/ner.py CHANGED
@@ -25,9 +25,9 @@ class NER:
25
 
26
  inputs = self.tokenizer(
27
  text,
28
- truncation=settings.TRUNCATE, # Enable truncation
29
- max_length=settings.MAX_LENGTH, # Set maximum length
30
- return_tensors="pt" # Make sure the tokenized inputs are returned as PyTorch tensors
31
  )
32
 
33
  pred = self.model(**inputs)
 
25
 
26
  inputs = self.tokenizer(
27
  text,
28
+ truncation=settings.TRUNCATE,
29
+ max_length=settings.MAX_LENGTH,
30
+ return_tensors="pt"
31
  )
32
 
33
  pred = self.model(**inputs)