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