diff --git a/transcribe_demo.py b/transcribe_demo.py index f8312c4..1a26a2b 100644 --- a/transcribe_demo.py +++ b/transcribe_demo.py @@ -16,7 +16,7 @@ def main(): parser = argparse.ArgumentParser() parser.add_argument("--model", default="medium", help="Model to use", - choices=["tiny", "base", "small", "medium", "large"]) + choices=["tiny", "base", "small", "medium", "large", "turbo"]) parser.add_argument("--non_english", action='store_true', help="Don't use the english model.") parser.add_argument("--energy_threshold", default=1000, @@ -63,7 +63,7 @@ def main(): # Load / Download model model = args.model - if args.model != "large" and not args.non_english: + if (args.model != "large" and args.model != "turbo") and not args.non_english: model = model + ".en" audio_model = whisper.load_model(model)