Update README.md

README.md

@@ -65,13 +65,15 @@ tokenized_query = ViTokenizer.tokenize(query)
 tokenized_sentences = [ViTokenizer.tokenize(sent) for sent in sentences]
 
 tokenized_pairs = [[tokenized_query, sent] for sent in tokenized_sentences]
+
+model_id = 'itdainb/PhoRanker'
 ```
 
 ## Usage with sentence-transformers
 
 ```python
 from sentence_transformers import CrossEncoder
-model = CrossEncoder('itdainb/vietnamese-cross-encoder', max_length=256)
+model = CrossEncoder(model_id, max_length=256)
 
 # For fp16 usage
 model.model.half()
@@ -84,8 +86,8 @@ scores = model.predict(tokenized_pairs)
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-model = AutoModelForSequenceClassification.from_pretrained('itdainb/vietnamese-cross-encoder')
-tokenizer = AutoTokenizer.from_pretrained('itdainb/vietnamese-cross-encoder')
+model = AutoModelForSequenceClassification.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 # For fp16 usage
 model.half()