Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import numpy as np
 import os
 import requests
+import spaces
+
 from fireredtts.fireredtts import FireRedTTS
 
 
@@ -31,6 +33,7 @@ tts = FireRedTTS(
     pretrained_path='pretrained_models',
 )
 
+@spaces.GPU
 def tts_inference(text, prompt_wav='examples/prompt_1.wav', lang='zh'):
     # Model inference
     syn_audio = tts.synthesize(