Update app.py

app.py

@@ -170,27 +170,43 @@ If you would like to use our V2 Model with GPU, then go to this [link](https://c
 
 API_OPEN = os.getenv('SPACE_ID') != 'hexgrad/Kokoro-TTS'
 API_NAME = None if API_OPEN else False
-with gr.Blocks() as app:
+with gr.Blocks(css=custom_css) as app:
     with gr.Row():
         gr.Markdown(BANNER_TEXT, container=True)
     with gr.Row():
         with gr.Column():
-            text = gr.Textbox(label='Input Text', info=f"Up to ~500 characters per Generate, or {'∞' if CHAR_LIMIT is None else CHAR_LIMIT} characters per Stream")
+            text = gr.Textbox(
+                label='Input Text', 
+                info=f"Up to ~500 characters per Generate, or {'∞' if CHAR_LIMIT is None else CHAR_LIMIT} characters per Stream"
+            )
             with gr.Row():
-                voice = gr.Dropdown(list(CHOICES.items()), value='af_heart', label='Voice', info='Quality and availability vary by language')
-                use_gpu = gr.Dropdown(
-                    [('ZeroGPU 🚀', True), ('CPU 🐌', False)],
-                    value=CUDA_AVAILABLE,
-                    label='Hardware',
-                    info='GPU is usually faster, but has a usage quota',
-                    interactive=CUDA_AVAILABLE
+                voice = gr.Dropdown(
+                    list(CHOICES.items()), 
+                    value='af_heart', 
+                    label='Voice', 
+                    info='Choose from our wide range of natural-sounding voices'
                 )
-            speed = gr.Slider(minimum=0.5, maximum=2, value=1, step=0.1, label='Speed')
-            random_btn = gr.Button('Random Text', variant='secondary')
+                # Remove the GPU choice dropdown and always use GPU if available
+                use_gpu = CUDA_AVAILABLE  # This will be used internally
+            speed = gr.Slider(
+                minimum=0.5, 
+                maximum=2, 
+                value=1, 
+                step=0.1, 
+                label='Speech Rate'
+            )
+            random_btn = gr.Button('Sample Text', variant='secondary')
         with gr.Column():
-            gr.TabbedInterface([generate_tab], ['Generate'])
+            gr.TabbedInterface([generate_tab], ['Generate Audio'])
+
+    # Keep the button click handlers but remove use_gpu from the visible inputs
     random_btn.click(fn=get_random_text, inputs=[voice], outputs=[text], api_name=API_NAME)
-    generate_btn.click(fn=generate_first, inputs=[text, voice, speed, use_gpu], outputs=[out_audio, out_ps], api_name=API_NAME)
+    generate_btn.click(
+        fn=generate_first, 
+        inputs=[text, voice, speed], 
+        outputs=[out_audio, out_ps], 
+        api_name=API_NAME
+    )
     tokenize_btn.click(fn=tokenize_first, inputs=[text, voice], outputs=[out_ps], api_name=API_NAME)
     predict_btn.click(fn=predict, inputs=[text, voice, speed], outputs=[out_audio], api_name=API_NAME)