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