diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 3cbbf815d65..1fa3d89bb5c 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -101,6 +101,7 @@ from .pipelines import ( PipelineDataFormat, QuestionAnsweringPipeline, TextClassificationPipeline, + TokenClassificationPipeline, pipeline, ) from .tokenization_albert import AlbertTokenizer diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py index b0ddcb919a4..170425d227d 100755 --- a/src/transformers/pipelines.py +++ b/src/transformers/pipelines.py @@ -649,6 +649,9 @@ class NerPipeline(Pipeline): return answers +TokenClassificationPipeline = NerPipeline + + class QuestionAnsweringArgumentHandler(ArgumentHandler): """ QuestionAnsweringPipeline requires the user to provide multiple arguments (i.e. question & context) to be mapped