diff --git a/transcribe.py b/transcribe.py index 4771884..d501687 100644 --- a/transcribe.py +++ b/transcribe.py @@ -1,11 +1,21 @@ import sys from unittest.mock import MagicMock -# Workaround for missing _lzma in some Python builds +# Comprehensive workaround for missing _lzma in some Python builds try: import lzma except ImportError: + mock_lzma = MagicMock() + # Add common constants that libraries expect from lzma + mock_lzma.FORMAT_XZ = 1 + mock_lzma.FORMAT_ALONE = 2 + mock_lzma.FORMAT_RAW = 3 + mock_lzma.CHECK_NONE = 0 + mock_lzma.CHECK_CRC32 = 1 + mock_lzma.CHECK_CRC64 = 4 + mock_lzma.CHECK_SHA256 = 10 sys.modules["_lzma"] = MagicMock() + sys.modules["lzma"] = mock_lzma import mlx_whisper import numpy as np