hatespeech / app.py
sandbox338's picture
Update app.py
5d2e259 verified
raw
history blame contribute delete
975 Bytes
import gradio as gr
from transformers import pipeline
# Load model from Hugging Face Hub
classifier = pipeline("text-classification", model="sandbox338/hatespeech")
# Map model labels to readable labels
label_map = {
"LABEL_0": "Non-hate speech",
"LABEL_1": "Political hate speech",
"LABEL_2": "Offensive language"
}
# Classification function
def classify_text(text):
result = classifier(text)[0]
label = result['label']
return label_map.get(label, "Unknown")
# Example inputs for testing
examples = [
["Hii ni ujumbe wa kawaida bila matusi."],
["Wanasiasa hawa ni wabaya na lazima waondoke!"],
["Unasema upuuzi na wewe ni mjinga kabisa!"]
]
# Gradio Interface
interface = gr.Interface(
fn=classify_text,
inputs=gr.Textbox(lines=4, placeholder="Andika maandishi ya Kiswahili hapa..."),
outputs="text",
title="Swahili Hate Speech Classifier",
examples=examples
)
if __name__ == "__main__":
interface.launch()