Update app.py

app.py

@@ -16,6 +16,9 @@ from faster_whisper import WhisperModel
 from moviepy.editor import VideoFileClip
 from transformers import AutoTokenizer, AutoModel
 
+import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 # ========================================
 #             Model Initialization
 # ========================================