emanuelaboros commited on
Commit
a4c6322
·
1 Parent(s): 18ee040

checking if other model thigns are needed

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. lang_detect.py +1 -1
config.json CHANGED
@@ -11,7 +11,7 @@
11
  },
12
  "custom_pipelines": {
13
  "lang-detect": {
14
- "impl": "lang_detect.MultitaskTokenClassificationPipeline",
15
  "pt": "AutoModelForTokenClassification"
16
  }
17
  },
 
11
  },
12
  "custom_pipelines": {
13
  "lang-detect": {
14
+ "impl": "lang_detect.LangDetectionPipeline",
15
  "pt": "AutoModelForTokenClassification"
16
  }
17
  },
lang_detect.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import Pipeline
2
 
3
 
4
- class MultitaskTokenClassificationPipeline(Pipeline):
5
 
6
  def _sanitize_parameters(self, **kwargs):
7
  preprocess_kwargs = {}
 
1
  from transformers import Pipeline
2
 
3
 
4
+ class LangDetectionPipeline(Pipeline):
5
 
6
  def _sanitize_parameters(self, **kwargs):
7
  preprocess_kwargs = {}