File size: 672 Bytes
addf785 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from transformers import AutoModel, AutoTokenizer, pipeline
from transformers.pipelines import PIPELINE_REGISTRY
from hive_token_classification import HiveTokenClassification
PIPELINE_REGISTRY.register_pipeline(
"hive-token-classification",
pipeline_class=HiveTokenClassification,
pt_model=AutoModel
)
pipe = pipeline(task='hive-token-classification', model='Hiveurban/dictabert-large-parse', trust_remote_code=True)
print(pipe(["讘砖谞转 1948 讛砖诇讬诐 讗驻专讬诐 拽讬砖讜谉 讗转 诇讬诪讜讚讬讜 讘驻讬住讜诇 诪转讻转 讜讘转讜诇讚讜转 讛讗诪谞讜转 讜讛讞诇 诇驻专住诐 诪讗诪专讬诐 讛讜诪讜专讬住讟讬讬诐"]))
pipe.push_to_hub('hive-token-classification')
|