v0

app.py

@@ -0,0 +1,15 @@
+import gradio as gr
+import transformers
+from transformers import pipeline
+
+model_name = "lsb/wikipedia-protected-classes"
+model_name = "lsb/test_trainer" # lol no time to get it right
+
+pipe = pipeline("text-classification", model_name)
+
+def predict(text):
+    return "🛡️ Protected" if pipe(text)[0]['label'] == "LABEL_1" else "✅ General Interest"
+
+iface = gr.Interface(fn=predict, inputs="text", outputs="text")
+iface.launch()
+