Update README.md

README.md

@@ -1,5 +1,7 @@
 ---
 pipeline_tag: text-classification
+metrics:
+- accuracy
 ---
 # BERT Text Classification Model
 
@@ -12,4 +14,6 @@ To use the model, you can call the `classify_text` function with a text input, a
 ```python
 text = "This is a positive review."
 predicted_class = classify_text(text)
-print("Predicted class:", predicted_class)
+print("Predicted class:", predicted_class)
+
+Devlin, J., Chang, M. W., Lee, K., & Toutanova, K. (2018). BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding. arXiv preprint arXiv:1810.04805.