register pipeline

register.py

@@ -0,0 +1,16 @@
+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')