This commit is contained in:
Timothy J. Baek
2024-02-14 23:32:54 -08:00
parent 8ab0338e71
commit 63e6f39b31
3 changed files with 8 additions and 9 deletions

View File

@@ -136,4 +136,6 @@ CHUNK_OVERLAP = 100
####################################
# Transcribe
####################################
WHISPER_MODEL_NAME = "base"
WHISPER_MODEL = os.getenv("WHISPER_MODEL", "base")
WHISPER_MODEL_DIR = os.getenv("WHISPER_MODEL_DIR", f"{CACHE_DIR}/whisper/models")