From fd63da542ef742e197ff7df304527f90ef7baa37 Mon Sep 17 00:00:00 2001 From: D3fense Date: Sat, 22 Jul 2023 20:49:41 +0200 Subject: [PATCH] Add lang selection in UI Whispers settings --- "ui/pages/05_\342\232\231\357\270\217_Settings.py" | 3 +++ 1 file changed, 3 insertions(+) diff --git "a/ui/pages/05_\342\232\231\357\270\217_Settings.py" "b/ui/pages/05_\342\232\231\357\270\217_Settings.py" index 3626982..e4a3849 100644 --- "a/ui/pages/05_\342\232\231\357\270\217_Settings.py" +++ "b/ui/pages/05_\342\232\231\357\270\217_Settings.py" @@ -51,6 +51,8 @@ ) task_options = ["transcribe", "translate"] task = st.selectbox("Default mode", options=task_options, index=task_options.index(fb.models.whisper.task)) + lang_options = ['en', 'zh', 'de', 'es', 'ru', 'ko', 'fr', 'ja', 'pt', 'tr', 'pl', 'ca', 'nl', 'ar', 'sv', 'it', 'id', 'hi', 'fi', 'vi', 'he', 'uk', 'el', 'ms', 'cs', 'ro', 'da', 'hu', 'ta', 'no', 'th', 'ur', 'hr', 'bg', 'lt', 'la', 'mi', 'ml', 'cy', 'sk', 'te', 'fa', 'lv', 'bn', 'sr', 'az', 'sl', 'kn', 'et', 'mk', 'br', 'eu', 'is', 'hy', 'ne', 'mn', 'bs', 'kk', 'sq', 'sw', 'gl', 'mr', 'pa', 'si', 'km', 'sn', 'yo', 'so', 'af', 'oc', 'ka', 'be', 'tg', 'sd', 'gu', 'am', 'yi', 'lo', 'uz', 'fo', 'ht', 'ps', 'tk', 'nn', 'mt', 'sa', 'lb', 'my', 'bo', 'tl', 'mg', 'as', 'tt', 'haw', 'ln', 'ha', 'ba', 'jw', 'su'] + language = st.selectbox("Language", options=lang_options, index=lang_options.index(fb.models.whisper.language)) save_config = st.form_submit_button(label="💾 Save") success_container = st.empty() @@ -65,5 +67,6 @@ fb.models.whisper.compression_ratio_threshold = compression_ratio_threshold fb.models.whisper.condition_on_previous_text = condition_on_previous_text fb.models.whisper.task = task + fb.models.whisper.language = language fb.models.save_settings() success_container.success("Settings saved!")