Update README.md

README.md

@@ -27,10 +27,10 @@ For more details, you can see our **paper** or [**GitHub**](https://github.com/A
 
 1. Download model by hugging face transformers.
 ```python
-from transformers import BartTokenizer, BartForConditionalGeneration
+from transformers import BertTokenizer, BertForMaskedLM, pipeline
 
-tokenizer = BartTokenizer.from_pretrained("AndyChiang/cdgp-csg-bert-dgen")
-csg_model = BartForConditionalGeneration.from_pretrained("AndyChiang/cdgp-csg-bert-dgen")
+tokenizer = BertTokenizer.from_pretrained("AndyChiang/cdgp-csg-bert-dgen")
+csg_model = BertForMaskedLM.from_pretrained("AndyChiang/cdgp-csg-bert-dgen")
 ```
 
 2. Create a unmasker.