kadirnar commited on
Commit
4ce3fcd
1 Parent(s): 607ca91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -6,8 +6,11 @@ from whisperplus.utils.download_utils import download_and_convert_to_mp3
6
  from whisperplus.utils.text_utils import format_speech_to_dialogue
7
 
8
  import subprocess
9
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
10
 
 
 
 
 
11
 
12
  def youtube_url_to_text(url, model_id, language_choice):
13
  """
 
6
  from whisperplus.utils.text_utils import format_speech_to_dialogue
7
 
8
  import subprocess
 
9
 
10
+ def install_package(package):
11
+ subprocess.check_call(['pip', 'install', package, '--no-build-isolation'])
12
+
13
+ install_package('flash-attn')
14
 
15
  def youtube_url_to_text(url, model_id, language_choice):
16
  """