Update app.py

app.py

@@ -45,29 +45,18 @@ from egogpt.conversation import conv_templates, SeparatorStyle
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 from huggingface_hub import snapshot_download
-
-# 下载整个模型文件夹到本地 ./llava-onevision-qwen2-7b-ov
-# snapshot_download(
-#     repo_id="lmms-lab/llava-onevision-qwen2-7b-ov",
-#     local_dir="./llava-onevision-qwen2-7b-ov",  # 指定本地存储目录
-#     ignore_patterns=["*.md", "*.txt"]  # 可以忽略一些不必要的文件（可选）
-# )
-
 from huggingface_hub import hf_hub_download
 
 # Download the model checkpoint file (large-v3.pt)
 ego_gpt_path = hf_hub_download(
-    repo_id="EgoLife-v1/speech_encoder",
-    filename="large-v3.pt",
-    local_dir="./"
+    repo_id="lmms-lab/EgoGPT-0.5b-Demo",
+    filename="./speech_encoder/large-v3.pt",
+    local_dir="./",
+    repo_type="model"
 )
 
 
-# pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
-# pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"
-# pretrained = 'EgoLife-v1/EgoGPT'
-pretrained = 'EgoLife-v1/EgoGPT-0.5b-Demo'
-# pretrained = "/mnt/sfs-common/jkyang/EgoGPT_release/checkpoints/EgoGPT-7b-Demo"
+pretrained = "lmms-lab/EgoGPT-0.5b-Demo"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 device_map = torch.device('cuda' if torch.cuda.is_available() else 'cpu')