Update README.md

README.md

@@ -1,6 +1,13 @@
+---
+license: gpl-3.0
+language:
+- en
+base_model:
+- google-bert/bert-base-uncased
+---
 # PatentBERT - PyTorch
 
-BERT model specialized for patent classification using the **real CPC (Cooperative Patent Classification) system** from the original PatentBERT training data.
+BERT model specialized for patent classification using the **real CPC (Cooperative Patent Classification) system**. (PyTorch version of the original [PatentBert](https://github.com/jiehsheng/PatentBERT/) model.)
 
 ## 📊 Specifications
 
@@ -85,3 +92,13 @@ This model was trained on a large patent corpus to automatically classify docume
 ## 📝 Citation
 
 If you use this model, please cite the original PatentBERT work and mention this PyTorch conversion.
+```
+@article{patent_bert, 
+  author = "Jieh-Sheng Lee and Jieh Hsiang",
+  title = "{PatentBERT: Patent classification with fine-tuning a pre-trained BERT model}",
+  journal = "World Patent Information",
+  volume = "61",
+  number = "101965",
+  year = "2020",
+}
+```