Ver código fonte

Use pip version of Whisper

Closes #26 #25
Miguel Piedrafita 2 anos atrás
pai
commit
8a572fc3b3
3 arquivos alterados com 3 adições e 11 exclusões
  1. 1 4
      auto_subtitle/cli.py
  2. 1 2
      requirements.txt
  3. 1 5
      setup.py

+ 1 - 4
auto_subtitle/cli.py

@@ -47,9 +47,6 @@ def main():
     if srt_only:
         return
 
-    # bash command to download a youtube video with `youtube-dl` and save it as `video.mp4`:
-    # youtube-dl -f 22 -o video.mp4 https://www.youtube.com/watch?v=QH2-TGUlwu4
-
     for path, srt_path in subtitles.items():
         out_path = os.path.join(output_dir, f"{filename(path)}.mp4")
 
@@ -58,7 +55,7 @@ def main():
         video = ffmpeg.input(path)
         audio = video.audio
 
-        stderr = ffmpeg.concat(
+        ffmpeg.concat(
             video.filter('subtitles', srt_path, force_style="OutlineColour=&H40000000,BorderStyle=3"), audio, v=1, a=1
         ).output(out_path).run(quiet=True, overwrite_output=True)
 

+ 1 - 2
requirements.txt

@@ -1,2 +1 @@
-youtube-dl
-git+https://github.com/openai/whisper.git
+openai-whisper

+ 1 - 5
setup.py

@@ -1,6 +1,3 @@
-import os
-
-import pkg_resources
 from setuptools import setup, find_packages
 
 setup(
@@ -10,8 +7,7 @@ setup(
     py_modules=["auto_subtitle"],
     author="Miguel Piedrafita",
     install_requires=[
-        'youtube-dl',
-        'whisper @ git+https://github.com/openai/whisper.git@main#egg=whisper'
+        'openai-whisper',
     ],
     description="Automatically generate and embed subtitles into your videos",
     entry_points={