Spaces:
Runtime error
Runtime error
import gradio as gr | |
from setfit import SetFitModel | |
model = SetFitModel.from_pretrained("johnpaulbin/beanbox-toxic") | |
def inf(inpt): | |
out = model.predict_proba([inpt]) | |
if out[0][0] > out[0][1]: | |
return "Not toxic" | |
else: | |
return "Toxic!" | |
iface = gr.Interface(fn=inf, inputs="text", outputs="text") | |
iface.queue(concurrency_count=500).launch() |