Update app.py

app.py

@@ -1,32 +1,22 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the model from Hugging Face Hub
+# Load the model
 classifier = pipeline("text-classification", model="sandbox338/hatespeech")
 
-# Define the label-to-number mapping
-label_to_number = {
-    "Non-hate speech": 0,
-    "Political hate speech": 1,
-    "Offensive language": 2
-}
-
-# Define the function for the interface
+# Define function to return only the label
 def classify_text(text):
-    results = classifier(text)
-    # Get the label and map it to the corresponding number
-    label = results[0]['label']
-    return label_to_number[label]
+    result = classifier(text)[0]
+    return result['label'] 
 
-# Create the interface
+# Gradio interface
 interface = gr.Interface(
     fn=classify_text,
     inputs=gr.Textbox(lines=4, placeholder="Enter Swahili text here..."),
     outputs="text",
     title="Swahili Hate Speech Classifier",
-    description="Classifies text as Non-hate speech, Political hate speech, or Offensive language."
+    description="Classifies text as one of: Non-hate speech, Political hate speech, or Offensive language."
 )
 
-# Launch the interface (only used if running locally)
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()