diff --git a/setup.py b/setup.py index 16c6570..ef26596 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ from setuptools import setup, find_packages setup( name="whisperlivekit", - version="0.1.9", + version="0.2.1", description="Real-time, Fully Local Whisper's Speech-to-Text and Speaker Diarization", long_description=open("README.md", "r", encoding="utf-8").read(), long_description_content_type="text/markdown", @@ -34,6 +34,7 @@ setup( }, package_data={ 'whisperlivekit': ['web/*.html'], + 'whisperlivekit.simul_whisper': ['dual_license_simulstreaming.md'], }, entry_points={ 'console_scripts': [ diff --git a/whisperlivekit/whisper_streaming_custom/backends.py b/whisperlivekit/whisper_streaming_custom/backends.py index 954c9d4..1b08c4e 100644 --- a/whisperlivekit/whisper_streaming_custom/backends.py +++ b/whisperlivekit/whisper_streaming_custom/backends.py @@ -14,9 +14,9 @@ from whisperlivekit.timed_objects import ASRToken logger = logging.getLogger(__name__) try: - from simul_whisper.config import AlignAttConfig - from simul_whisper.simul_whisper import PaddedAlignAttWhisper, DEC_PAD - from simul_whisper.whisper import tokenizer + from whisperlivekit.simul_whisper.config import AlignAttConfig + from whisperlivekit.simul_whisper.simul_whisper import PaddedAlignAttWhisper, DEC_PAD + from whisperlivekit.simul_whisper.whisper import tokenizer SIMULSTREAMING_AVAILABLE = True except ImportError: logger.warning("SimulStreaming dependencies not available. SimulStreaming backend will not be available.") @@ -315,7 +315,7 @@ class SimulStreamingASR(ASRBase): def __init__(self, lan, modelsize=None, cache_dir=None, model_dir=None, logfile=sys.stderr, **kwargs): if not SIMULSTREAMING_AVAILABLE: - raise ImportError("SimulStreaming dependencies are not available. Please install SimulStreaming requirements.") + raise ImportError("""SimulStreaming dependencies are not available. Please install WhisperLiveKit using pip install "whisperlivekit[simulstreaming]". If you are building from source, you should also copy the content of the simul_whisper directory from the SimulStreaming repository into whisperlivekit/simul_whisper.""") with open("whisperlivekit/simul_whisper/dual_license_simulstreaming.md", "r") as f: print("*"*80 + f.read() + "*"*80) self.logfile = logfile