Update README.md

README.md

@@ -1,3 +1,6 @@
+---
+license: bigscience-openrail-m
+---
 # SecureBERT: A Domain-Specific Language Model for Cybersecurity
 SecureBERT is a domain-specific language model based on RoBERTa which is trained on a huge amount of cybersecurity data and fine-tuned/tweaked to understand/represent cybersecurity textual data.
 
@@ -80,4 +83,15 @@ while True:
     predict_mask(sent, tokenizer, model)
      
     print("===========================\n")
-```
+```
+# Reference
+@inproceedings{aghaei2023securebert, 
+title={SecureBERT: A Domain-Specific Language Model for Cybersecurity}, 
+author={Aghaei, Ehsan and Niu, Xi and Shadid, Waseem and Al-Shaer, Ehab}, 
+booktitle={Security and Privacy in Communication Networks: 
+18th EAI International Conference, SecureComm 2022, Virtual Event, 
+October 2022, 
+Proceedings}, 
+pages={39--56}, 
+year={2023}, 
+organization={Springer} }