Update README.md

README.md

@@ -1,3 +1,40 @@
----
-license: cc0-1.0
----
+---
+license: cc0-1.0
+datasets:
+- kairaamilanii/cyberbullying-indonesia
+language:
+- id
+metrics:
+- accuracy
+- confusion_matrix
+base_model:
+- indolem/indobertweet-base-uncased
+pipeline_tag: text-classification
+---
+
+This model is based on a BERT model trained with a few bullying detection datasets. It is trained exclusively in the Indonesian language.
+
+```python
+from transformers import BertTokenizer, AutoModelForSequenceClassification
+
+model_path = 'kairaamilanii/IndoBERT-Bullying-Classifier'
+tokenizer = BertTokenizer.from_pretrained(model_path)
+model = AutoModelForSequenceClassification.from_pretrained(model_path)
+
+text = "KOK JELEK BANGET SIH" # Example text for prediction
+
+inputs = tokenizer(text, return_tensors="pt")
+with torch.no_grad():
+    outputs = model(**inputs)
+    predicted_class = torch.argmax(outputs.logits, dim=-1).item()
+print(f"Predicted class: {predicted_class}")
+
+if predicted_class == 1:
+    print("Prediction: Bullying")
+else:
+    print("Prediction: Non-bullying")
+```
+example output:
+```python
+[{'Predicted class': 1, 'Prediction': Bullying}]
+```