File size: 512 Bytes
2048275 6ab4d6a 2048275 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from transformers import pipeline, AutoModelForSequenceClassification
from transformers.pipelines import PIPELINE_REGISTRY
from bert_paper_classifier import SciBertPaperClassifierPipeline
PIPELINE_REGISTRY.register_pipeline(
"paper-classification",
pipeline_class=SciBertPaperClassifierPipeline,
pt_model=AutoModelForSequenceClassification,
)
pipe = pipeline(task="paper-classification", model="HibiscusMaximus/scibert_paper_classification")
pipe.push_to_hub("bert-paper-classification-pipeline")
|