tuhailong commited on
Commit
37f7a4d
1 Parent(s): c5dfe30

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -14,7 +14,10 @@ model is [simcse](https://arxiv.org/abs/2104.08821).
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
  ```
 
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)
23
  ```