improved warning message for English-only models

MiscellaneousStuff · Sep 25, 2022 · 2d3032d · 2d3032d
1 parent 8cf36f3
commit 2d3032d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/whisper/transcribe.py b/whisper/transcribe.py
@@ -257,7 +257,7 @@ def cli():
  device: str = args.pop("device")
  os.makedirs(output_dir, exist_ok=True)
 
- if model_name.endswith(".en") and args["language"] != "en":
+ if model_name.endswith(".en") and args["language"] not in {"en", "English"}:
  warnings.warn(f"{model_name} is an English-only model but receipted '{args['language']}'; using English instead.")
  args["language"] = "en"