m3hrdadfi commited on
Commit
c9c39e4
·
1 Parent(s): d630b49

Fix pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ class TypoDetector:
38
  if not self.debug:
39
  self.config = AutoConfig.from_pretrained(self.model_name_or_path)
40
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
41
- self.model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_path, config=config)
42
  self.nlp = pipeline(
43
  self.task_name,
44
  model=self.model,
 
38
  if not self.debug:
39
  self.config = AutoConfig.from_pretrained(self.model_name_or_path)
40
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
41
+ self.model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_path, config=self.config)
42
  self.nlp = pipeline(
43
  self.task_name,
44
  model=self.model,