diff --git a/Buzz.spec b/Buzz.spec index c06a319b..4a9339ff 100644 --- a/Buzz.spec +++ b/Buzz.spec @@ -1,4 +1,18 @@ # -*- mode: python ; coding: utf-8 -*- +from PyInstaller.utils.hooks import collect_data_files, copy_metadata + +datas = [] +datas += collect_data_files('torch') +datas += copy_metadata('tqdm') +datas += copy_metadata('torch') +datas += copy_metadata('tqdm') +datas += copy_metadata('regex') +datas += copy_metadata('requests') +datas += copy_metadata('packaging') +datas += copy_metadata('filelock') +datas += copy_metadata('numpy') +datas += copy_metadata('tokenizers') +datas += collect_data_files('whisper') block_cipher = None @@ -8,8 +22,9 @@ a = Analysis( ['main.py'], pathex=[], binaries=[], - datas=[], - hiddenimports=[], + datas=datas, + hiddenimports=['googleapiclient', 'apiclient', 'pytorch', '“sklearn.utils._cython_blas”', + '“sklearn.neighbors.typedefs”', '“sklearn.neighbors.quad_tree”', '“sklearn.tree”', '“sklearn.tree._utils”'], hookspath=[], hooksconfig={}, runtime_hooks=[], diff --git a/Makefile b/Makefile index 4c8aaea4..5856b28b 100644 --- a/Makefile +++ b/Makefile @@ -1,2 +1,2 @@ buzz: - pyinstaller --name 'Buzz' --windowed main.py + pyinstaller --noconfirm Buzz.spec diff --git a/transcriber.py b/transcriber.py index 04864dd9..0ba75168 100644 --- a/transcriber.py +++ b/transcriber.py @@ -41,6 +41,7 @@ class Transcriber: result = self.model.transcribe( audio=chunk_path, language="en") + # TODO: this should probably be a callback or output buffer print(result["text"]) os.remove(chunk_path)