Update app.py

app.py

@@ -5,6 +5,7 @@ import argparse
 import torch
 import gradio as gr
 import numpy as np
+import spaces
 
 import ChatTTS
 
@@ -21,7 +22,7 @@ def generate_seed():
         "__type__": "update",
         "value": new_seed
     }
-
+@spaces.GPU
 def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
     torch.manual_seed(audio_seed_input)
     rand_spk = torch.randn(768)