From: F1L1P Date: Wed, 6 Mar 2024 22:25:10 +0000 (+0100) Subject: examples : Auto lowercase language parameter in main.cpp (#1928) X-Git-Tag: upstream/1.7.4~933 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=2e2626b1677161d055531aa9b4d7172a148a7138;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp examples : Auto lowercase language parameter in main.cpp (#1928) * Auto lowercase language parameter * Update examples/main/main.cpp Co-authored-by: bobqianic --------- Co-authored-by: bobqianic --- diff --git a/examples/main/main.cpp b/examples/main/main.cpp index cef96306..84db6a0c 100644 --- a/examples/main/main.cpp +++ b/examples/main/main.cpp @@ -82,6 +82,14 @@ struct whisper_params { void whisper_print_usage(int argc, char ** argv, const whisper_params & params); +char* whisper_param_turn_lowercase(char* in){ + int string_len = strlen(in); + for(int i = 0; i < string_len; i++){ + *(in+i) = tolower((unsigned char)*(in+i)); + } + return in; +} + bool whisper_params_parse(int argc, char ** argv, whisper_params & params) { for (int i = 1; i < argc; i++) { std::string arg = argv[i]; @@ -135,7 +143,7 @@ bool whisper_params_parse(int argc, char ** argv, whisper_params & params) { else if (arg == "-pc" || arg == "--print-colors") { params.print_colors = true; } else if (arg == "-pp" || arg == "--print-progress") { params.print_progress = true; } else if (arg == "-nt" || arg == "--no-timestamps") { params.no_timestamps = true; } - else if (arg == "-l" || arg == "--language") { params.language = argv[++i]; } + else if (arg == "-l" || arg == "--language") { params.language = whisper_param_turn_lowercase(argv[++i]); } else if (arg == "-dl" || arg == "--detect-language") { params.detect_language = true; } else if ( arg == "--prompt") { params.prompt = argv[++i]; } else if (arg == "-m" || arg == "--model") { params.model = argv[++i]; }