Create app.py

app.py

@@ -0,0 +1,16 @@
+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.launch()