kadirnar commited on
Commit
22dcbc5
1 Parent(s): 8d029a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -15,6 +15,9 @@ from whisperplus.pipelines.text2speech import TextToSpeechPipeline
15
  from whisperplus.pipelines.whisper_autocaption import WhisperAutoCaptionPipeline
16
  from whisperplus.pipelines.whisper_diarize import ASRDiarizationPipeline
17
 
 
 
 
18
 
19
  def youtube_url_to_text(url, model_id, language_choice):
20
  """
 
15
  from whisperplus.pipelines.whisper_autocaption import WhisperAutoCaptionPipeline
16
  from whisperplus.pipelines.whisper_diarize import ASRDiarizationPipeline
17
 
18
+ import subprocess
19
+
20
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
21
 
22
  def youtube_url_to_text(url, model_id, language_choice):
23
  """