Update model size with fetching from env

ClubCedille · Jul 18, 2024 · c297a82 · c297a82
1 parent b6b4918
commit c297a82
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/point.py b/point.py
@@ -15,8 +15,7 @@
 
 logger = logging.getLogger("point")
 
-model_size = "base"
-
+model_size = os.getenv("WHISPER_MODEL_SIZE", "base")
 
 def transcribe(audio_file: io.BytesIO) -> Iterator[str]:
     header = time.strftime("%Y-%m-%d %H:%M:%S") + f", transcript with {model_size}"