Update app.py

app.py

@@ -14,13 +14,12 @@ from internvl.model.internvl_chat import InternVLChatModel
 
 # 模型配置
 CHECKPOINTS = {
-    "TokenOCR-4096-English-seg": "/path/to/TokenOCR_4096_English_seg",
-    "TokenOCR-2048-Bilingual-seg": "/path/to/TokenOCR_2048_Binlinual_seg",
-    "R50":"model/checkpoint.pth",
-    "R50_siglip": "/path/to/R50_siglip_checkpoint.pth"
+    "TokenFD-4096-English-seg": "TongkunGuan/TokenFD_4096_English_seg",
+    "TokenFD-2048-Bilingual-seg": "TongkunGuan/TokenFD_2048_Bilingual_seg",
 }
 
 # 全局变量
+HF_TOKEN = os.getenv("HF_TOKEN")
 current_vis = []
 current_bpe = []
 current_index = 0
@@ -42,7 +41,7 @@ def load_model(check_type):
 
     elif 'TokenOCR' in check_type:
         model_path = CHECKPOINTS[check_type]
-        tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, use_fast=False)
+        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()
         transform = T.Compose([
             T.Lambda(lambda img: img.convert('RGB')),