Fix issue in model download

aalramadan · Jul 11, 2024 · 7d74666 · 7d74666
1 parent 0d38161
commit 7d74666
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/compile/build.txt b/compile/build.txt
@@ -1,5 +1,5 @@
 Build (Windows):
-pyinstaller --windowed --hidden-import=tqdm --hidden-import=faster-whisper --icon="C:\Users\Ali\Desktop\TransVisio\images\images\icon.ico" main.py
+pyinstaller --windowed --icon="C:\Users\Ali\Desktop\TransVisio\images\images\icon.ico" main.py
 
 Build (Mac OS):
 pyinstaller --onefile --windowed --icon="/Users/Ali/Desktop/TransVisio/images/images/icon.icns" main.py

diff --git a/ui_functions.py b/ui_functions.py
@@ -230,7 +230,7 @@ def extract_text_from_video(main_window, selected_file, api_key_whisper):
         return output 
     else:
         device = 'gpu' if cuda.is_available() else 'cpu'
-        model = WhisperModel(main_window.combo_model_size.currentText(), device=device)
+        model = WhisperModel(main_window.combo_model_size.currentText(), device=device, download_root=path.dirname(selected_file))
         output, info = model.transcribe(filename)
         srt_string = ""
         for i, segment in enumerate(output):