update

app_locally.py

@@ -14,6 +14,13 @@ ckpt_converter = 'checkpoints/converter'
 tone_color_converter = ToneColorConverter(f'{ckpt_converter}/config.json', device=device)
 tone_color_converter.load_ckpt(f'{ckpt_converter}/checkpoint.pth')
 
+
+languages = ["EN", "ES", "FR", "ZH", "JP", "KR"]
+en = ["EN-Default", "EN-US", "EN-BR", "EN_INDIA", "EN-AU"]
+
+LANG = sys.argv[1].strip()
+
+
 #def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
 def predict(prompt, audio_file_pth, mic_file_path, use_mic, language):
     # initialize a empty info
@@ -179,7 +186,10 @@ with gr.Blocks(analytics_enabled=False) as demo:
                 info="Notice: Microphone input may not work properly under traffic",
             )
             #language = gr.Radio(['EN-Newest', 'EN-US', 'EN-BR', 'EN_INDIA', 'EN-AU', 'EN-Default', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN-Newest')
-            language = gr.Radio(['EN-US', 'EN-BR', 'EN_INDIA', 'EN-AU', 'EN-Default', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN-US')
+            if LANG == "EN":
+                language = gr.Radio(['EN-US', 'EN-BR', 'EN_INDIA', 'EN-AU', 'EN-Default'], label='Language', value='EN-US')
+            else:
+                language = gr.Radio([LANG], value=LANG)
 
             tts_button = gr.Button("Send", elem_id="send-btn", visible=True)