Merge pull request #725 from jnkstr/no-internet-whisper

fix: no internet connection for whisper if you use docker
2024-02-15 02:44:31 -05:00
parent c0b5e66df6 63e6f39b31
commit 1def55cf09
3 changed files with 13 additions and 5 deletions
@@ -139,4 +139,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")