diff --git a/transcribe_demo.py b/transcribe_demo.py index 81e918a..6dd8972 100644 --- a/transcribe_demo.py +++ b/transcribe_demo.py @@ -16,7 +16,7 @@ from sys import platform def main(): parser = argparse.ArgumentParser() - parser.add_argument("--model", default="tiny", help="Model to use", + parser.add_argument("--model", default="medium", help="Model to use", choices=["tiny", "base", "small", "medium", "large"]) parser.add_argument("--non_english", action='store_true', help="Don't use the english model.")