from transformers import pipeline
classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model", top_k=None)
classifier("This is a modern sentence!")

json

[[
{'label': 'modern', 'score': 0.901931643486023},
{'label': 'shakespearean', 'score': 0.09806833416223526}
]]

Return top label

python

from transformers import pipeline
classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model")
classifier("This is a modern sentence!")

json

[
{'label': 'modern', 'score': 0.901931643486023}
]