diff --git a/management/commands/transcribe.py b/management/commands/transcribe.py index 4350251..8bf1e95 100644 --- a/management/commands/transcribe.py +++ b/management/commands/transcribe.py @@ -21,8 +21,6 @@ class Command(BaseCommand): default=None, help='import into layer (default subtitle layer)') parser.add_argument('--translate', type=str, dest='translate', default='', help='list of languages to translate: (i.e. hi:en,de:en') - parser.add_argument('--model', type=str, dest='model', - default='small', help='whisper model (small, large-v3-turbo, default: small)') parser.add_argument('--gpu', action='store_true', dest='gpu', default=False, help='user GPU (default: disabled)') parser.add_argument('--join-sentences', action='store_true', dest='join_sentences', diff --git a/transcribe.py b/transcribe.py index fcce57f..b750f8c 100644 --- a/transcribe.py +++ b/transcribe.py @@ -92,7 +92,7 @@ def prepare_joint_annotations(result, target_length=200): return annotations -def extract_subtitles(item, user, layer, translate, gpu=False, join_sentences=False, model="small"): +def extract_subtitles(item, user, layer, translate, gpu=False, join_sentences=False): language = None if "language" not in item.data: language = None @@ -108,7 +108,6 @@ def extract_subtitles(item, user, layer, translate, gpu=False, join_sentences=Fa run_py = os.path.join(os.path.dirname(os.path.abspath(__file__)), "run_whisper.py") cmd = ["/opt/whisper-timestamped/bin/python", run_py] - cmd += ["--model", model] if language: cmd += ["--language", language] if translate and language in translate: @@ -149,7 +148,7 @@ def extract_subtitles(item, user, layer, translate, gpu=False, join_sentences=Fa return True -def extract_subtitles_cmd(item, user, layer, translate, gpu=False, model="small"): +def extract_subtitles_cmd(item, user, layer, translate, gpu=False): if "language" not in item.data: language = None else: @@ -160,7 +159,7 @@ def extract_subtitles_cmd(item, user, layer, translate, gpu=False, model="small" src = item.streams()[0].media.path tmp = tempfile.mkdtemp() - cmd = ["/opt/whisper-timestamped/bin/whisper_timestamped", "--model", model] + cmd = ["/opt/whisper-timestamped/bin/whisper_timestamped", "--model", "turbo"] if language: cmd += ["--language", language] if translate and language in translate: @@ -241,7 +240,6 @@ def main(**kwargs): translate, kwargs.get("gpu"), join_sentences=kwargs.get("join_sentences"), - model=kwargs.get("model", "small") ): done.items.add(item) queue.items.remove(item)