emanuelaboros
/

lang-detect

Token Classification

language-identification

Model card Files Files and versions

emanuelaboros commited on Feb 27

Commit

3041710

·

1 Parent(s): a4c6322

name changning

Files changed (2) hide show

config.json +2 -2
modeling_lang.py +1 -1

config.json CHANGED Viewed

@@ -1,13 +1,13 @@
 {
   "_name_or_path": "emanuelaboros/lang-detect",
   "architectures": [
-    "ExtendedMultitaskModelForTokenClassification"
   ],
   "filename": "LID-40-3-2000000-1-4.bin",
   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_lang.ImpressoConfig",
-    "AutoModelForTokenClassification": "modeling_lang.ExtendedMultitaskModelForTokenClassification"
   },
   "custom_pipelines": {
     "lang-detect": {

 {
   "_name_or_path": "emanuelaboros/lang-detect",
   "architectures": [
+    "LangDetectorModel"
   ],
   "filename": "LID-40-3-2000000-1-4.bin",
   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_lang.ImpressoConfig",
+    "AutoModelForTokenClassification": "modeling_lang.LangDetectorModel"
   },
   "custom_pipelines": {
     "lang-detect": {

modeling_lang.py CHANGED Viewed

@@ -8,7 +8,7 @@ from .configuration_lang import ImpressoConfig
 logger = logging.getLogger(__name__)
-class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
     config_class = ImpressoConfig
     def __init__(self, config):

 logger = logging.getLogger(__name__)
+class LangDetectorModel(PreTrainedModel):
     config_class = ImpressoConfig
     def __init__(self, config):