Update README.md

README.md

@@ -1,14 +1,22 @@
-#Data
+---
+language: zh
+tags:
+- simcse
+datasets:
+- dialogue
+---
+
+# Data
 unsupervise train data is E-commerce dialogue.
 
-##Model
+## Model
 model is [simcse](https://arxiv.org/abs/2104.08821). 
 
-##Usage
+### Usage
 ```python
->>>from transformers import AutoTokenizer, AutoModel
->>>model = AutoModel.from_pretrained("tuhailong/simcse_model")
->>>tokenizer = AutoTokenizer.from_pretrained("tuhailong/simcse_model")
->>>sentences_str_list = ["今天天气不错的","天气不错的"]
->>>inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
->>>outputs = model(**inputs)
+>>> from transformers import AutoTokenizer, AutoModel
+>>> model = AutoModel.from_pretrained("tuhailong/simcse_model")
+>>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/simcse_model")
+>>> sentences_str_list = ["今天天气不错的","天气不错的"]
+>>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
+>>> outputs = model(**inputs)