def main()

in whisperx-model-fetch/download_whisperx_models.py [0:0]

• 17 lines of code
• 6 McCabe index (conditional complexity)

def main(
        whisper_models: bool = typer.Option(False, help="Download whisper models"),
        diarization_models: bool = typer.Option(False, help="Download diarization models"),
        torch_align_models: bool = typer.Option(False, help="Download torch align models"),
        huggingface_align_models: bool = typer.Option(False, help="Download huggingface align models"),
        huggingface_token: str = typer.Option("", help="Huggingface authentication token")):
    if whisper_models:
        download_all_whisper_models()
    if diarization_models:
        if not huggingface_token:
            print("Please provide a Huggingface authentication token (--huggingface-token <token>)")
            sys.exit(1)
        download_diarization_models(huggingface_token)
    if torch_align_models:
        download_torch_align_models()
    if huggingface_align_models:
        download_huggingface_align_models()