Hugging Face's logo

AndyChiang
/
cdgp-csg-bert-cloth

Fill-Mask
Transformers
PyTorch
English
bert
cloze
distractor
generation
Inference Endpoints
Model card Files Community
1
AndyChiang commited on
Commit
204b87c
•
1 Parent(s): 01eb1b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ For more details, you can see our **paper** or [**GitHub**](https://github.com/A
27
 
28
  1. Download the model by hugging face transformers.
29
  ```python
30
- from transformers import BertTokenizer, BertForMaskedLM
31
 
32
  tokenizer = BertTokenizer.from_pretrained("AndyChiang/cdgp-csg-bert-cloth")
33
  csg_model = BertForMaskedLM.from_pretrained("AndyChiang/cdgp-csg-bert-cloth")
 
27
 
28
  1. Download the model by hugging face transformers.
29
  ```python
30
+ from transformers import BertTokenizer, BertForMaskedLM, pipeline
31
 
32
  tokenizer = BertTokenizer.from_pretrained("AndyChiang/cdgp-csg-bert-cloth")
33
  csg_model = BertForMaskedLM.from_pretrained("AndyChiang/cdgp-csg-bert-cloth")