Update app.py for v1

app.py

@@ -34,10 +34,11 @@ from egogpt.constants import (
     DEFAULT_SPEECH_TOKEN
 )
 from egogpt.conversation import conv_templates, SeparatorStyle
-
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 # 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 = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"
+pretrained = '/EgoLife-v1/EgoGPT'
 device = "cuda"
 device_map = "cuda"
 
@@ -91,7 +92,8 @@ bibtext = """
 ```
 """
 
-cur_dir = os.path.dirname(os.path.abspath(__file__))
+# cur_dir = os.path.dirname(os.path.abspath(__file__))
+cur_dir = ''
 
 
 def time_to_frame_idx(time_int: int, fps: int) -> int:

requirements.txt

@@ -1,7 +1,11 @@
-gradio==5.12.0
-gradio_client==1.5.4
-open_clip_torch==2.29.0
-torch==2.1.2
-torchaudio==2.1.2
-torchvision==0.16.2
-transformers==4.45.2
+torch>=2.0.0
+numpy>=1.23.0
+uvicorn>=0.22.0
+librosa>=0.10.0
+whisper>=1.0.0
+requests>=2.31.0
+fastapi>=0.95.0
+pydantic>=1.10.0
+decord>=0.6.0
+gradio>=3.0.0
+transformers>=4.30.0