Update README.md

README.md

@@ -1,3 +1,10 @@
+---
+language:
+- en
+base_model:
+- s-nlp/roberta_toxicity_classifier
+pipeline_tag: text-classification
+---
 # Student Chat Toxicity Classifier
 
 This model is a fine-tuned version of the `s-nlp/roberta_toxicity_classifier` and is designed to classify text-based messages in student conversations as **toxic** or **non-toxic**. It is specifically tailored to detect and flag malpractice suggestions, unethical advice, or any toxic communication while encouraging ethical and positive interactions among students.
@@ -71,4 +78,4 @@ def predict_toxicity(text):
 # Test the model
 message = "You can copy answers during the exam."
 prediction = predict_toxicity(message)
-print(f"Message: {message}\nPrediction: {prediction}")
+print(f"Message: {message}\nPrediction: {prediction}")