Update app.py

app.py

@@ -23,7 +23,8 @@ def _init_kokoro() -> None:
     if KModel is None or KPipeline is None:
         raise gr.Error("Kokoro is not installed. Please add 'kokoro>=0.9.4' and 'torch' to requirements and install.")
     
-    device = "cpu"
+    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)}
     try:
@@ -141,7 +142,7 @@ def generate_tts(text: str, speed: float, voice: str):
 
 
 with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
-    gr.HTML("<h1 style='text-align: center;'>TTS-Hub</h1><p style='text-align: center;'>Powered by Kokoro</p>")
+    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()
     default_kokoro_voice = (