Update app.py

app.py

@@ -25,8 +25,8 @@ def _init_kokoro() -> None:
     
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Using device: {device}")
-    model = KModel().to(device).eval()
-    pipelines = {"a": KPipeline(lang_code="a", model=False)}
+    model = KModel(repo_id="hexgrad/Kokoro-82M").to(device).eval()
+    pipelines = {"a": KPipeline(lang_code="a", model=False, repo_id="hexgrad/Kokoro-82M")}
     try:
         pipelines["a"].g2p.lexicon.golds["kokoro"] = "kˈOkəɹO"
     except Exception:
@@ -141,7 +141,7 @@ def generate_tts(text: str, speed: float, voice: str):
     yield from kokoro_tts_stream(text, speed, voice)
 
 
-with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
+with gr.Blocks() as demo:
     gr.HTML("<h1 style='text-align: center;'>Kokoro-TTS</h1><p style='text-align: center;'>Powered by Kokoro-82M on CPU</p>")
 
     available_voices = get_kokoro_voices()
@@ -179,7 +179,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
         label="Generated Speech",
         streaming=True,
         autoplay=True,
-        show_download_button=True,
+        buttons=["download"],
     )
 
     generate_inputs = [text_input, kokoro_speed, kokoro_voice]
@@ -198,4 +198,5 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
         api_name="generate_speech_enter"
     )
 
-demo.queue().launch(debug=True)
+if __name__ == "__main__":
+    demo.queue().launch(debug=True, theme='Nymbo/Nymbo_Theme')