Update app.py

app.py

@@ -36,13 +36,21 @@ from egogpt.constants import (
 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 hf_hub_download
+from huggingface_hub import snapshot_download
 
-# 下载模型 'llava-onevision-qwen2-7b-ov' 到本地目录
-hf_hub_download(repo_id="lmms-lab/llava-onevision-qwen2-7b-ov", cache_dir="./llava-onevision-qwen2-7b-ov")
+# 下载整个模型文件夹到本地 ./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"]  # 可以忽略一些不必要的文件（可选）
+)
 
-# 下载模型 'large-v3.pt' 到本地目录
-hf_hub_download(repo_id="EgoLife-v1/EgoGPT/large-v3.pt", cache_dir="./large-v3.pt")
+from huggingface_hub import hf_hub_download
+egogpt_path = hf_hub_download(
+    repo_id="EgoLife-v1/EgoGPT", 
+    filename="large-v3.pt",
+    cache_dir="./large-v3.pt"
+)
 
 
 # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"