test: download model before running

chidiwilliams · Mar 14, 2024 · eb1d56c · eb1d56c
1 parent 3e546cc
commit eb1d56c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/buzz/cli.py b/buzz/cli.py
@@ -4,7 +4,12 @@
 
 from PyQt6.QtCore import QCommandLineParser, QCommandLineOption
 
-from buzz.model_loader import ModelType, WhisperModelSize, TranscriptionModel
+from buzz.model_loader import (
+    ModelType,
+    WhisperModelSize,
+    TranscriptionModel,
+    ModelDownloader,
+)
 from buzz.store.keyring_store import KeyringStore
 from buzz.transcriber.transcriber import (
     Task,
@@ -148,6 +153,7 @@ def parse(app: Application, parser: QCommandLineParser):
             whisper_model_size=model_size,
             hugging_face_model_id=hugging_face_model_id,
         )
+        ModelDownloader(model=model).run()
         model_path = model.get_local_model_path()
 
         if model_path is None: