Update app.py

app.py

@@ -16,7 +16,7 @@ if not IS_DUPLICATE:
 
 CHAR_LIMIT = None if IS_DUPLICATE else 5000
 models = {gpu: KModel().to('cuda' if gpu else 'cpu').eval() for gpu in [False] + ([True] if CUDA_AVAILABLE else [])}
-pipelines = {lang_code: KPipeline(lang_code=lang_code, model=False) for lang_code in 'abefhip'}
+pipelines = {lang_code: KPipeline(lang_code=lang_code, model=False) for lang_code in 'ab'}
 pipelines['a'].g2p.lexicon.golds['kokoro'] = 'kˈOkəɹO'
 pipelines['b'].g2p.lexicon.golds['kokoro'] = 'kˈQkəɹQ'
 
@@ -68,6 +68,7 @@ def tokenize_first(text, voice='af_heart'):
     return ''
 
 def generate_all(text, voice='af_heart', speed=1, use_gpu=CUDA_AVAILABLE):
+    text = f"[_](/{indo_to_ipa(text)}/)"
     text = text if CHAR_LIMIT is None else text.strip()[:CHAR_LIMIT]
     pipeline = pipelines[voice[0]]
     pack = pipeline.load_voice(voice)
@@ -135,11 +136,11 @@ CHOICES = {
 '🇬🇧 🚹 Fable': 'bm_fable',
 '🇬🇧 🚹 Lewis': 'bm_lewis',
 '🇬🇧 🚹 Daniel': 'bm_daniel',
-'Espanol':'ef_dora',
-'Hindi':'hf_alpha',
-'French':'ff_siwis',
-'Italian':'if_sara',
-'Portuguese':'pf_dora',
+# 'Espanol':'ef_dora',
+# 'Hindi':'hf_alpha',
+# 'French':'ff_siwis',
+# 'Italian':'if_sara',
+# 'Portuguese':'pf_dora',
 }
 for v in CHOICES.values():
     pipelines[v[0]].load_voice(v)
@@ -154,6 +155,8 @@ TOKEN_NOTE = '''
 ⬆️ Raise stress 1 level `[or](+2)` 2 levels (only works on less stressed, usually short words)
 '''
 
+TOKEN_NOTE = ""
+
 with gr.Blocks() as generate_tab:
     out_audio = gr.Audio(label='Output Audio', interactive=False, streaming=False, autoplay=True)
     generate_btn = gr.Button('Generate', variant='primary')
@@ -181,7 +184,7 @@ with gr.Blocks() as stream_tab:
 BANNER_TEXT = '''
 [***Kokoro*** **is an open-weight TTS model with 82 million parameters.**](https://huggingface.co/hexgrad/Kokoro-82M)
 
-This demo showcases Indonesian with English accent, but you can use various accents of other languages.
+This demo showcases Indonesian with English accent
 '''
 API_OPEN = os.getenv('SPACE_ID') != 'hexgrad/Kokoro-TTS'
 API_NAME = None if API_OPEN else False