tuhailong commited on
Commit
c5dfe30
1 Parent(s): 1ef9b5a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -6
README.md CHANGED
@@ -14,9 +14,7 @@ model is [simcse](https://arxiv.org/abs/2104.08821).
14
 
15
  ### Usage
16
  ```python
17
- >>> from transformers import AutoTokenizer, AutoModel
18
- >>> model = AutoModel.from_pretrained("tuhailong/simcse_model")
19
- >>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/simcse_model")
20
- >>> sentences_str_list = ["今天天气不错的","天气不错的"]
21
- >>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
22
- >>> outputs = model(**inputs)
 
14
 
15
  ### Usage
16
  ```python
17
+ >>> from transformers import pipeline
18
+ >>> unmasker = pipeline('fill-mask', model='bert-base-uncased')
19
+ >>> unmasker("Hello I'm a [MASK] model.")
20
+ ```