Update app.py

app.py

@@ -36,8 +36,14 @@ 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)
-subprocess.run('huggingface-cli download lmms-lab/llava-onevision-qwen2-7b-ov --cache-dir ./llava-onevision-qwen2-7b-ov')
-subprocess.run('huggingface-cli download EgoLife-v1/EgoGPT/large-v3.pt --cache-dir ./large-v3.pt')
+from huggingface_hub import hf_hub_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")
+
+# 下载模型 'large-v3.pt' 到本地目录
+hf_hub_download(repo_id="EgoLife-v1/EgoGPT/large-v3.pt", cache_dir="./large-v3.pt")
+
 
 # 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"