You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'd love to be able to use a trained SetFit model, e.g.:
importspacyimportspacy_setfit# Load the spaCy language model:nlp=spacy.load("en_core_web_sm")
# Add the "text_categorizer" pipeline component to the spaCy modelnlp.add_pipe("text_categorizer", config={
"pretrained_model_name_or_path": "addy88/sst5-setfit-model",
})
doc=nlp("I really love that film.")
print(doc.cats)
Right now this throws an error.
Tom Aarsen
The text was updated successfully, but these errors were encountered:
I'd love to be able to use a trained SetFit model, e.g.:
Right now this throws an error.
The text was updated successfully, but these errors were encountered: