gptsubtitler

Legendador

Legende automaticamente qualquer vídeo falado em qualquer idioma para um idioma de sua escolha usando IA.

Modelos usados:

• Porta OpenAI Whisper C++ - para áudio para texto
• Facebook M2M10 - para tradução

Ferramentas usadas:

• ffmpeg

Não se esqueça de marcar o repositório com estrela se você achar que é útil ou educativo!

Antes:

soldado.mp4

Depois (em romeno - model_type=medium, language_model_type=base ):

soldado_legendado.mp4

Configurar

Instale usando pip .

 pip install gptsubtitler

Instale ffmpeg :

 # Ubuntu or Debian
sudo apt update && sudo apt install ffmpeg

# MacOS
brew install ffmpeg

# Windows using Chocolatey https://chocolatey.org/
choco install ffmpeg

Guia rápido

Exemplo de uso para adicionar legendas e traduzi-las em romeno:

Linha de comando:

gptsubtitler soldier.mp4 --source_language en --target_language ro --captioning_model_type medium --language_model_type base

Ou em Python

 from gptsubtitler import Transcriber

# I strongly recommend using the "medium" model_type.
Transcriber . transcribe ( "soldier.mp4" , source_language = "en" , target_language = "ro" , captioning_model_type = "medium" , language_model_type = "base" )

Você também pode usar a classe Translator de translator.py se quiser apenas traduzir algum texto.

Exemplo de uso para tradução de inglês para romeno:

 from gptsubtitler import Translator

print ( Translator . translate ( "Hi!" , target_language = "ro" , source_language = "en" ))

Se você gerou um arquivo .srt e deseja apenas adicionar legendas:

 from gptsubtitler import create_video_with_subtitles
create_video_with_subtitles ( "video.mp4" , "output.srt" , "video_subtitled.mp4" )

Opções

 Args:
    video_file (str): Path to video file.

    output_video_file (str, optional): Path to output video file. Defaults to video_file_subtitled.

    output_subtitle_file (str, optional): Path to output SRT file. Defaults to "output.srt".

    source_language (str, optional): Source language for translation. Defaults to en.

    target_language (str, optional): Target language for translation. Defaults to None.

    captioning_model_type (str, optional): Model type. Defaults to "base".

    language_model_type (str, optional): Language model type. Defaults to "base".

    model_dir (str, optional): Path to model directory. Defaults to None.

Opções disponíveis para captioning_model_type (o modelo de áudio para texto):

• pequeno
• base - padrão
• pequeno
• médio
• grande

Opções disponíveis para language_model_type (o modelo de tradutor de idioma):

• base - padrão
• grande