Update Modules/Generate_Speech.py

Modules/Generate_Speech.py

@@ -68,8 +68,8 @@ def _init_kokoro() -> None:
                 device = "cuda"
         except Exception:
             device = "cpu"
-    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:
@@ -169,7 +169,7 @@ def build_interface() -> gr.Interface:
                 info="Select from 54 available voices across multiple languages and accents",
             ),
         ],
-        outputs=gr.Audio(label="Audio", type="numpy", format="wav", show_download_button=True),
+        outputs=gr.Audio(label="Audio", type="numpy", format="wav", buttons=["download"]),
         title="Generate Speech",
         description=(
             "<div style=\"text-align:center\">Generate speech with Kokoro-82M. Supports multiple languages and accents. Runs on CPU or CUDA if available.</div>"