Spaces:
Runtime error
Runtime error
File size: 371 Bytes
9f86c43 |
1 2 3 4 5 6 7 8 |
from transformers.pipelines import PIPELINE_REGISTRY,AutoModel,AutoConfig
from models.bert import BertCrfModel,BertCrfConfig
from pipelines import NERPredictorPipe
def register():
PIPELINE_REGISTRY.register_pipeline("ner_predictor", pipeline_class=NERPredictorPipe)
AutoConfig.register("bert_crf",BertCrfConfig)
AutoModel.register(BertCrfConfig,BertCrfModel) |