restructured config

config.json

@@ -1,12 +1,31 @@
-
-{"custom_pipelines": {
-  "language-detection": {
-    "impl": "impresso_langident_wrapper.Pipeline_One",
-    "pt": [
-      "AutoModelForSequenceClassification"
-    ],
-    "tf": [
-    ]
-  }
+{
+  "_name_or_path": "Maslionok/pipeline1",
+  "architectures": [
+    "BertForTokenClassification"
+  ],
+  "custom_pipelines": {
+      "language-detection": {
+          "impl": "impresso_langident_wrapper.Pipeline_One",
+          "pt": ["AutoModel"],
+          "tf": []
+      }
+  },
+  "id2label": {
+      "0": "English",
+      "1": "German",
+      "2": "French"
+  },
+  "label2id": {
+      "English": 0,
+      "German": 1,
+      "French": 2
+  },
+  "model_type": "bert",
+  "vocab_size": 2000000,  
+  "embedding_dim": 300,   
+  "hash_count": 4,        
+  "minn": 3,             
+  "maxn": 6,             
+  "bucket": 2000000,      
+  "num_labels": 40
 }
-}

impresso_langident_wrapper.py

@@ -17,10 +17,13 @@ class Pipeline_One(Pipeline):
 
     def postprocess(self, outputs, **kwargs):
         return outputs
-    
 
-# Register the pipeline
+
+from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification
+
 PIPELINE_REGISTRY.register_pipeline(
     "language-detection",
     pipeline_class=Pipeline_One,
+    pt_model=AutoModelForSequenceClassification,
+    tf_model=TFAutoModelForSequenceClassification,
 )