nisten commited on
Commit
c290e7a
1 Parent(s): 9e0937f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,17 +6,17 @@ import sys
6
  from threading import Thread
7
  from transformers import AutoTokenizer, TextIteratorStreamer
8
  import numpy as np
9
- import librosa
10
  from queue import Queue
11
  from threading import Event
12
 
13
  # Install required packages
14
- subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "--force-reinstall", "--no-deps" "librosa", "parler_tts", "melotts", "funasr", "sounddevice", "nltk", "sounddevice", "deepfilternet", "einops", "accelerate", "ChatTTS", "git+https://github.com/Muennighoff/transformers.git@olmoe"])
15
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
16
 
17
  # Import the custom OlmoeForCausalLM
18
  from transformers import OlmoeForCausalLM
19
-
20
  # Import speech-to-speech components
21
  from VAD.vad_handler import VADHandler
22
  from STT.whisper_stt_handler import WhisperSTTHandler
 
6
  from threading import Thread
7
  from transformers import AutoTokenizer, TextIteratorStreamer
8
  import numpy as np
9
+
10
  from queue import Queue
11
  from threading import Event
12
 
13
  # Install required packages
14
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "--force-reinstall", "librosa", "parler_tts", "melotts", "funasr", "sounddevice", "nltk", "sounddevice", "deepfilternet", "einops", "accelerate", "ChatTTS", "git+https://github.com/Muennighoff/transformers.git@olmoe"])
15
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
16
 
17
  # Import the custom OlmoeForCausalLM
18
  from transformers import OlmoeForCausalLM
19
+ import librosa
20
  # Import speech-to-speech components
21
  from VAD.vad_handler import VADHandler
22
  from STT.whisper_stt_handler import WhisperSTTHandler