How to use davanstrien/finepdfs-edu-purpose-classifier with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("davanstrien/finepdfs-edu-purpose-classifier") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds)
3055b70
1
2
3
4
{ "module_input_name": "sentence_embedding", "module_output_name": "sentence_embedding" }
