MirrorMirror
/
01
mirror of https://github.com/OpenInterpreter/01.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: download whisper model directly from huggingface

This commit is contained in:
Shiven Mian 2024-02-10 20:06:06 -08:00
parent 9384f68c66
commit 169a750644
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
OS/01/start.sh
View File

@ -1,9 +1,11 @@
### SETTINGS ### SETTINGS
# If ALL_LOCAL is False, we'll use OpenAI's services # If ALL_LOCAL is False, we'll use OpenAI's services
# If setting ALL_LOCAL to true, set the path to the WHISPER local model # else we use whisper.cpp and piper local models
export ALL_LOCAL=False export ALL_LOCAL=False
# export WHISPER_MODEL_PATH=... export WHISPER_MODEL_NAME="ggml-tiny.en.bin"
# Set the OpenAI API key for OpenInterpreter to work
# export OPENAI_API_KEY=sk-... # export OPENAI_API_KEY=sk-...
# For TTS, we use the en_US-lessac-medium voice model by default # For TTS, we use the en_US-lessac-medium voice model by default
@ -32,6 +34,11 @@ export LOG_LEVEL="INFO"
### SETUP ### SETUP
# if using local models, install the models / executables # if using local models, install the models / executables
WHISPER_MODEL_URL="https://huggingface.co/ggerganov/whisper.cpp/resolve/main/"
WHISPER_RUST_PATH="`pwd`/local_stt/whisper-rust"
curl -OL "${WHISPER_MODEL_URL}${WHISPER_MODEL_NAME}" --output-dir ${WHISPER_RUST_PATH}
if [[ "$ALL_LOCAL" == "True" ]]; then if [[ "$ALL_LOCAL" == "True" ]]; then
OS=$(uname -s) OS=$(uname -s)
ARCH=$(uname -m) ARCH=$(uname -m)

11
OS/01/stt.py
View File

@ -53,13 +53,14 @@ def run_command(command):
return result.stdout, result.stderr return result.stdout, result.stderr
def get_transcription_file(wav_file_path: str): def get_transcription_file(wav_file_path: str):
model_path = os.getenv("WHISPER_MODEL_PATH") whisper_rust_path = os.path.join(os.path.dirname(__file__), 'local_stt', 'whisper-rust')
if not model_path: model_name = os.getenv('WHISPER_MODEL_NAME')
raise EnvironmentError("WHISPER_MODEL_PATH environment variable is not set.") if not model_name:
raise EnvironmentError("WHISPER_MODEL_NAME environment variable is not set.")
output, error = run_command([ output, error = run_command([
os.path.join(os.path.dirname(__file__), 'local_stt', 'whisper-rust', 'whisper-rust'), os.path.join(whisper_rust_path, 'whisper-rust'),
'--model-path', model_path, '--model-path', os.path.join(whisper_rust_path, model_name),
'--file-path', wav_file_path '--file-path', wav_file_path
]) ])