Update app.py

app.py

@@ -98,12 +98,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)
 
 
 @spaces.GPU
@@ -174,11 +174,11 @@ CSS ="""
 
 demo = gr.ChatInterface(
     fn=chat,
-    description="""Try [Vintern-1B](https://huggingface.co/5CD-AI/Viet-InternVL2-1B) in this demo. Vintern-1B consists of [InternViT-300M-448px](https://huggingface.co/OpenGVLab/InternViT-300M-448px), an MLP projector, and [Qwen2-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2-0.5B-Instruct).""",
+    description="""Try [Vintern-1B-v2](https://huggingface.co/5CD-AI/Viet-InternVL2-1B) in this demo. Vintern-1B-v2 consists of [InternViT-300M-448px](https://huggingface.co/OpenGVLab/InternViT-300M-448px), an MLP projector, and [Qwen2-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2-0.5B-Instruct).""",
     examples=[{"text": "Mô tả hình ảnh.", "files":["./demo_3.jpg"]},
               {"text": "Trích xuất các thông tin từ ảnh.", "files":["./demo_1.jpg"]}, 
               {"text": "Mô tả hình ảnh một cách chi tiết.", "files":["./demo_2.jpg"]}],
-    title="❄️ Vintern-1B ❄️",
+    title="❄️ Vintern-1B-v2 ❄️",
     multimodal=True,
     css=CSS
 )