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 **CPC (Cooperative Patent Classification) system**. ([PatentBert](https://github.com/jiehsheng/PatentBERT) converted into PyTorch version)
 
 ## 📊 Specifications
 
@@ -57,7 +64,7 @@ predicted_class_id = predictions.argmax().item()
 confidence = predictions.max().item()
 
 # Use model labels (real CPC codes)
-predicted_label = model.config.id2label[str(predicted_class_id)]
+predicted_label = model.config.id2label[predicted_class_id]
 
 print(f"Predicted CPC class: {predicted_label} (ID: {predicted_class_id})")
 print(f"Confidence: {confidence:.2%}")
@@ -80,8 +87,7 @@ This model was trained on a large patent corpus to automatically classify docume
 
 - [Cooperative Patent Classification (CPC)](https://www.cooperativepatentclassification.org/)
 - [Original PatentBERT Paper](https://arxiv.org/abs/2103.02557)
-- [Hugging Face Transformers](https://huggingface.co/transformers/)
 
 ## 📝 Citation
 
-If you use this model, please cite the original PatentBERT work and mention this PyTorch conversion.
+If you use this model, please cite the original PatentBERT work and mention this PyTorch conversion.