cuda + default

app.py

@@ -5,12 +5,15 @@ sys.path.append("neutts-air")
 from neuttsair.neutts import NeuTTSAir
 import gradio as gr
 
-# load model
+DEFAULT_REF_TEXT = "So I'm live on radio. And I say, well, my dear friend James here clearly, and the whole room just froze. Turns out I'd completely misspoken and mentioned our other friend." 
+DEFAULT_REF_PATH = os.path.join(os.getcwd(), "/neutts-air/samples/dave.wav")
+DEFAULT_GEN_TEXT = "Hello, I'm NeuTTS-Air! How're you doing today?"
+
 tts = NeuTTSAir(
     backbone_repo="neuphonic/neutts-air",
-    backbone_device="cpu",
+    backbone_device="cuda",
     codec_repo="neuphonic/neucodec",
-    codec_device="cpu"
+    codec_device="cuda"
 )
 
 @spaces.GPU()
@@ -28,11 +31,18 @@ def infer(ref_text, ref_audio_path, gen_text):
 demo = gr.Interface(
     fn=infer,
     inputs=[
-        gr.Textbox(label="Reference Text", value="So I'm live on radio. And I say, well, my dear friend James here clearly, and the whole room just froze. Turns out I'd completely misspoken and mentioned our other friend."),
-        gr.Audio(type="filepath", label="Reference Audio", value=os.path.join(os.getcwd(), "/neutts-air/samples/dave.wav")),
-        gr.Textbox(label="Text to Generate", value="My name is Dave, and um, I'm from London."),
+        gr.Textbox(label="Reference Text"),
+        gr.Audio(type="filepath", label="Reference Audio"),
+        gr.Textbox(label="Text to Generate"),
     ],
     outputs=gr.Audio(type="numpy", label="Generated Speech"),
+    examples=[
+        [
+            DEFAULT_REF_TEXT, 
+            DEFAULT_REF_PATH, 
+            DEFAULT_GEN_TEXT
+        ],
+    ],
     title="NeuTTS-Air☁️",
     description="Upload a reference audio sample, provide the reference text, and enter new text to synthesize."
 )