racism analysis

app.py

@@ -0,0 +1,32 @@
+
+import gradio as gr
+from transformers import pipeline
+
+RACISM_MODEL = "davidmasip/racism"
+racism_analysis_pipe = pipeline("text-classification",
+                                model=RACISM_MODEL, tokenizer=RACISM_MODEL)
+
+
+def racism_analysis(text):
+    results = racism_analysis_pipe(text)
+    return results[0]["label"], round(results[0]["score"], 5)
+
+
+gradio_ui = gr.Interface(
+    fn=racism_analysis,
+    title="Racism Detector (Spanish)",
+    description="Enter some text and check if model detects bullying.",
+    inputs=[
+        gr.inputs.Textbox(lines=5, label="Paste some text here"),
+    ],
+    outputs=[
+        gr.outputs.Textbox(label="Label"),
+        gr.outputs.Textbox(label="Score"),
+    ],
+    examples=[
+        ["Eres mas alto que un pino y mas tonto que un pepino!"],
+        ["Que divertida"]
+    ],
+)
+
+gradio_ui.launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch