Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
# Load the hate speech detection model from Hugging Face | |
classifier = pipeline("text-classification", model="unitary/unbiased-toxic-roberta") | |
def detect_hate_speech(text): | |
"""Detect hate speech in the given text.""" | |
sentences = text.split(".") | |
hate_speech = [] | |
for sentence in sentences: | |
if sentence.strip(): | |
result = classifier(sentence.strip())[0] | |
if result["label"] in ["toxic", "insult", "hate"]: | |
hate_speech.append(sentence.strip()) | |
return hate_speech if hate_speech else "No hate speech detected" | |
# Create Gradio interface | |
iface = gr.Interface( | |
fn=detect_hate_speech, | |
inputs=gr.Textbox(lines=5, placeholder="Enter text here..."), | |
outputs="json", | |
title="Hate Speech Detection", | |
description="Enter a sentence or paragraph to check for hate speech.", | |
) | |
# Launch the app | |
iface.launch() | |