app.py

app.py

@@ -4,11 +4,19 @@ from transformers import pipeline
 # Load the model from Hugging Face Hub
 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
 def classify_text(text):
     results = classifier(text)
-    # Return only the label without the score
-    return results[0]['label']
+    # Get the label and map it to the corresponding number
+    label = results[0]['label']
+    return label_to_number[label]
 
 # Create the interface
 interface = gr.Interface(