Update app.py

app.py

@@ -44,7 +44,7 @@ def load_model(check_type):
         model_path = CHECKPOINTS[check_type]
         tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, use_fast=False, use_auth_token=HF_TOKEN)
         # model = InternVLChatModel.from_pretrained(model_path, torch_dtype=torch.bfloat16).eval()
-        model = InternVLChatModel.from_pretrained(checkpoint_vit_english, low_cpu_mem_usage=True, torch_dtype=torch.bfloat16 ,load_in_8bit=False, load_in_4bit=False).eval()
+        model = InternVLChatModel.from_pretrained(model_path, low_cpu_mem_usage=True, torch_dtype=torch.bfloat16 ,load_in_8bit=False, load_in_4bit=False).eval()
         transform = get_transform(is_train=False, image_size=model.config.force_image_size)
         
     return model.to(device), tokenizer, transform, device