Spaces:
Sleeping
Sleeping
| 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() | |