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