kadirnar commited on
Commit
67bd832
·
verified ·
1 Parent(s): 1af2f60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -5,6 +5,9 @@ from whisperplus.pipelines.whisper_diarize import ASRDiarizationPipeline
5
  from whisperplus.utils.download_utils import download_and_convert_to_mp3
6
  from whisperplus.utils.text_utils import format_speech_to_dialogue
7
 
 
 
 
8
 
9
  def youtube_url_to_text(url, model_id, language_choice):
10
  """
 
5
  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
  """