Update app.py

app.py

@@ -4,7 +4,7 @@ import soundfile as sf
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from xcodec2.modeling_xcodec2 import XCodec2Model
 import tempfile
-
+import spaces
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 ####################
@@ -39,7 +39,7 @@ def extract_speech_ids(speech_tokens_str):
         else:
             print(f"Unexpected token: {token_str}")
     return speech_ids
-
+@spaces.GPU
 def text2speech(input_text, speaker_choice):
     """
     将文本转为语音波形，并返回音频文件路径
@@ -105,8 +105,8 @@ demo = gr.Interface(
     inputs=[gr.Textbox(label="Enter text", lines=5),
             gr.Dropdown(choices=speaker_choices, label="Select Speaker", value="puck")],
     outputs=gr.Audio(label="Generated Audio", type="filepath"),
-    title="Llasa-1B TTS Demo",
-    description="Input a piece of text in English or Chinese, select a speaker (puck or kore), and click to generate speech.\nModel: HKUST-Audio/Llasa-1B + HKUST-Audio/xcodec2"
+    title="Llasa-1B TTS finetuned using shb777/gemini-flash-2.0-speech",
+    description="Input a piece of text in English, select a speaker (puck or kore), and click to generate speech.\nModel: HKUSTAudio/Llasa-1B-two-speakers-kore-puck + HKUSTAudio/xcodec2"
 )
 
 if __name__ == "__main__":