| 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') | |