Update README.md

README.md

@@ -62,11 +62,11 @@ class SentenceEmbeddingPipeline(Pipeline):
         return sentence_embeddings
 
 # load optimized model
-onnx_path = "./models/cos-v1-best/"
-model = ORTModelForFeatureExtraction.from_pretrained(onnx_path, file_name="model_quantized.onnx")
+model_name = "rawsh/multi-qa-MiniLM-distill-onnx-L6-cos-v1"
+model = ORTModelForFeatureExtraction.from_pretrained(model_name, file_name="model_quantized.onnx")
 
 # create optimized pipeline
-tokenizer = AutoTokenizer.from_pretrained(onnx_path, use_fast=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 optimized_emb = SentenceEmbeddingPipeline(model=model, tokenizer=tokenizer)
 pred1 = optimized_emb("Hello world!")
 pred2 = optimized_emb("I hate everything.")