Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 14:35:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

transcribe() on English-only model won't complain when language="en" is not given

Browse Source
This commit is contained in:
Jong Wook Kim 2022-10-09 02:40:12 -07:00
parent 82725cea9c
commit d18e9ea5dd
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
whisper/transcribe.py
View File

@ -84,6 +84,9 @@ def transcribe(
mel = log_mel_spectrogram(audio) mel = log_mel_spectrogram(audio)
if decode_options.get("language", None) is None: if decode_options.get("language", None) is None:
if not model.is_multilingual:
decode_options["language"] = "en"
else:
if verbose: if verbose:
print("Detecting language using up to the first 30 seconds. Use `--language` to specify the language") print("Detecting language using up to the first 30 seconds. Use `--language` to specify the language")
segment = pad_or_trim(mel, N_FRAMES).to(model.device).to(dtype) segment = pad_or_trim(mel, N_FRAMES).to(model.device).to(dtype)
@ -282,6 +285,7 @@ def cli():
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
if model_name.endswith(".en") and args["language"] not in {"en", "English"}: if model_name.endswith(".en") and args["language"] not in {"en", "English"}:
if args["language"] is not None:
warnings.warn(f"{model_name} is an English-only model but receipted '{args['language']}'; using English instead.") warnings.warn(f"{model_name} is an English-only model but receipted '{args['language']}'; using English instead.")
args["language"] = "en" args["language"] = "en"