Update README.md

README.md

@@ -45,8 +45,8 @@ def mean_pooling(model_output, attention_mask):
 sentences = ['This is an example sentence', 'Each sentence is converted']
 
 # Load model from HuggingFace Hub
-tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
-model = ORTModelForFeatureExtraction.from_pretrained('sentence-transformers/all-mpnet-base-v2')
+tokenizer = AutoTokenizer.from_pretrained('yilunzhang/all-mpnet-base-v2-onnx')
+model = ORTModelForFeatureExtraction.from_pretrained('yilunzhang/all-mpnet-base-v2-onnx')
 
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')