Views
No views yet
!pip install setfit1from setfit import SetFitModel
2
3model = SetFitModel.from_pretrained("johnpaulbin/beanbox-toxic")
4
5inpt = "" #@param {type:"string"}
6
7out = model.predict_proba([inpt])
8
9if out[0][0] > out[0][1]:
10 print("Not toxic")
11else:
12 print("Toxic!")
13
14print(f"NOT TOXIC: {out[0][0]}\nTOXIC: {out[0][1]}")