Update README.md

README.md

@@ -85,8 +85,8 @@ def encode(text_or_image_list):
     embeddings = F.normalize(reps, p=2, dim=1).detach().cpu().numpy()
     return embeddings
 
-tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG", trust_remote_code=True)
-model = AutoModel.from_pretrained("openbmb/VisRAG", torch_dtype=torch.bfloat16, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("openbmb/VisRAG-Ret", trust_remote_code=True)
+model = AutoModel.from_pretrained("openbmb/VisRAG-Ret", torch_dtype=torch.bfloat16, trust_remote_code=True)
 model.eval()
 
 script_dir = os.path.dirname(os.path.realpath(__file__))
@@ -96,7 +96,7 @@ passages = [
     Image.open(os.path.join(script_dir, 'test_image/dog.jpg')).convert('RGB'),
 ]
 
-INSTRUCTION = "Represent this query for retrieving relavant documents: "
+INSTRUCTION = "Represent this query for retrieving relevant documents: "
 queries = [INSTRUCTION + query for query in queries]
 
 embeddings_query = encode(queries)