Views
No views yet
allenai/scibert_scivocab_uncased for document section text classificationpip install -U transformersfrom transformers import pipeline
model_tag = "ml4pubmed/scibert-scivocab-uncased_pub_section"
classifier = pipeline(
'text-classification',
model=model_tag,
)
prompt = """
Experiments on two machine translation tasks show these models to be superior in quality while being more parallelizable and requiring significantly less time to train.
"""
classifier(
prompt,
) # classify the sentence