JunchuanYu commited on
Commit
63d2dc2
1 Parent(s): a99defd

Update gramformer.py

Browse files
Files changed (1) hide show
  1. gramformer.py +1 -1
gramformer.py CHANGED
@@ -5,7 +5,7 @@ class Gramformer:
5
  from transformers import AutoModelForSeq2SeqLM
6
  #from lm_scorer.models.auto import AutoLMScorer as LMScorer
7
  import errant
8
- self.annotator = errant.load('en')
9
 
10
  if use_gpu:
11
  device= "cuda:0"
 
5
  from transformers import AutoModelForSeq2SeqLM
6
  #from lm_scorer.models.auto import AutoLMScorer as LMScorer
7
  import errant
8
+ self.annotator = errant.load('en_core_web_sm')
9
 
10
  if use_gpu:
11
  device= "cuda:0"