Update README.md

README.md

@@ -253,12 +253,12 @@ def load_image(image_file, input_size=448, max_num=12):
     return pixel_values
 
 model = AutoModel.from_pretrained(
-    "5CD-AI/Viet-InternVL2-1B",
+    "5CD-AI/Vintern-1B-v2",
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
     trust_remote_code=True,
 ).eval().cuda()
-tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Viet-InternVL2-1B", trust_remote_code=True, use_fast=False)
+tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Vintern-1B-v2", trust_remote_code=True, use_fast=False)
 
 test_image = 'test-image.jpg'