diff --git a/auto_subtitle/utils/ffmpeg.py b/auto_subtitle/utils/ffmpeg.py index 7e4e6f2..5ad0957 100644 --- a/auto_subtitle/utils/ffmpeg.py +++ b/auto_subtitle/utils/ffmpeg.py @@ -108,7 +108,7 @@ def hard_subtitles(input_path: str, output_path: str, force_style="OutlineColour=&H40000000,BorderStyle=3") if translated.tmp_file is not None: - intermediate.filter( + intermediate = intermediate.filter( 'subtitles', translated.tmp_file_path, force_style="OutlineColour=&H40000000,BorderStyle=3,Alignment=6") ffmpeg.concat( diff --git a/requirements.txt b/requirements.txt index 07456b2..55f0439 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,13 @@ -faster-whisper==1.0.2 +faster-whisper==1.0.3 tqdm==4.66.4 ffmpeg-python==0.2.0 langcodes==3.3.0 transformers>=4.4,<5 -torch>=1.6.0 +torch>=1.6.0,<=2.3.1 numpy>=1.24.2,<=1.26.4 -nltk -sentencepiece -protobuf +nltk<=3.8.1 +sentencepiece<=0.2.0 +protobuf<=5.27.2 +huggingface_hub<=0.23.5 # this package should be installed with `pip install --no-deps EasyNMT==2.0.2` # EasyNMT==2.0.2 \ No newline at end of file diff --git a/setup.py b/setup.py index 7dac23d..f6992a4 100644 --- a/setup.py +++ b/setup.py @@ -7,20 +7,21 @@ py_modules=["auto_subtitle"], author="Sergey Chernyaev", install_requires=[ - 'faster-whisper==1.0.2', + 'faster-whisper==1.0.3', 'tqdm==4.66.4', 'ffmpeg-python==0.2.0', 'langcodes==3.3.0', 'transformers>=4.4,<5', - 'torch>=1.6.0', + 'torch>=1.6.0,<=2.3.1', 'numpy>=1.24.2,<=1.26.4', - 'nltk', - 'sentencepiece', - 'protobuf', + 'nltk<=3.8.1', + 'sentencepiece<=0.2.0', + 'protobuf<=5.27.2', + 'huggingface_hub<=0.23.5', ], description="Automatically generate and embed subtitles into your videos", entry_points={ 'console_scripts': ['faster_auto_subtitle=auto_subtitle.cli:main'], }, include_package_data=True, -) +) \ No newline at end of file