How to use baansafe/s1-triage with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("baansafe/s1-triage") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds)
