Update README.md

README.md

@@ -1,10 +1,11 @@
 # 🧠 BERT Classifier for Black Article Detection
 
 ## 📝 Model Overview
-This repository hosts a fine-tuned BERT model (`bert-base-uncased`) for classifying newspaper articles based on whether they center on Black people. The training dataset is provided for full reproducibility.
+This repository hosts a fine-tuned BERT model (`bert-base-uncased`) for classifying newspaper articles based on their focus on Black people. The training dataset is also provided for reproducibility.
 
 ## 📖 Description
-The model was trained on 2,000 manually labeled sentences from historical newspaper articles (1960–1973). Labels indicate if the article primarily concerns Black people.
+- **Model:** Fine-tuned `bert-base-uncased`
+- **Training Data:** 2,000 manually labeled sentences from historical newspaper articles (1960–1973)
 - **Inputs:** `sentence` *(string)*
 - **Outputs:** `black_story` *(0 or 1)*
 
@@ -15,13 +16,39 @@ The model was trained on 2,000 manually labeled sentences from historical newspa
 - **Recall:** 92.1%
 
 ## 🚀 Usage Instructions
-To use this model via Hugging Face Transformers:
 ```python
 from transformers import pipeline
 classifier = pipeline("text-classification", model="mikemcrae/black-article-classifier")
 result = classifier("Black activists led a peaceful protest downtown.")
 print(result)
+```
 
+## 💾 Training Dataset
+- Hugging Face Dataset: [mikemcrae/black-article-training-data](https://huggingface.co/datasets/mikemcrae/black-article-training-data)
+- CSV Columns: `sentence`, `black_story`
 
+## ⚙️ Reproduction Instructions
+```python
+from datasets import load_dataset
+from transformers import BertForSequenceClassification, Trainer, TrainingArguments
+
+dataset = load_dataset("mikemcrae/black-article-training-data")
+model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=2)
+```
+
+## 📜 License
+**MIT License**: Free to use with attribution.
+```
 MIT License © 2025 Mike McRae
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND.
+```
+
+## ❤️ Citation
+```
+@inproceedings{mcrae2025blackbert,
+  title={BERT Classifier for Black Article Detection},
+  author={Mike McRae},
+  year={2025},
+  url={https://huggingface.co/mikemcrae/black-article-classifier}
+}
+```