diff --git a/examples/transcribe.py b/examples/transcribe.py index 73b505b..4fa1d39 100644 --- a/examples/transcribe.py +++ b/examples/transcribe.py @@ -4,7 +4,7 @@ voices_folder = "voices" # voices folder containing voice samples for recognition language = "en" # language code log_folder = "logs" # log folder for storing transcripts -modelSize = "medium" # size of model to be used [tiny, small, medium, large-v1, large-v2, large-v3] +modelSize = "tiny" # size of model to be used [tiny, small, medium, large-v1, large-v2, large-v3] quantization = False # setting this 'True' may speed up the process but lower the accuracy transcriptor = Transcriptor(file, log_folder, language, modelSize, voices_folder, quantization) diff --git a/setup.py b/setup.py index a2a7af5..07aba52 100644 --- a/setup.py +++ b/setup.py @@ -5,12 +5,12 @@ setup( name="speechlib", - version="1.0.11", + version="1.0.13", description="speechlib is a library that can do speaker diarization, transcription and speaker recognition on an audio file to create transcripts with actual speaker names. This library also contain audio preprocessor functions.", packages=find_packages(), long_description=long_description, long_description_content_type="text/markdown", - url="https://github.com/Navodplayer1/speechlib", + url="https://github.com/NavodPeiris/speechlib", author="Navod Peiris", author_email="navodpeiris1234@gmail.com", license="MIT", @@ -19,6 +19,6 @@ "Programming Language :: Python :: 3.10", "Operating System :: OS Independent", ], - install_requires=["transformers", "torch", "torchaudio", "pydub", "pyannote.audio", "speechbrain", "accelerate", "faster-whisper"], + install_requires=["transformers==4.36.2", "torch==2.1.2", "torchaudio==2.1.2", "pydub==0.25.1", "pyannote.audio==3.1.1", "speechbrain==0.5.16", "accelerate==0.26.1", "faster-whisper==0.10.0"], python_requires=">=3.8", ) \ No newline at end of file diff --git a/setup_instruction.md b/setup_instruction.md index 94ad9b6..1ed948d 100644 --- a/setup_instruction.md +++ b/setup_instruction.md @@ -9,7 +9,7 @@ for publishing: pip install twine for install locally for testing: - pip install dist/speechlib-1.0.10-py3-none-any.whl + pip install dist/speechlib-1.0.13-py3-none-any.whl finally run: twine upload dist/*