Update app.py

app.py

@@ -28,6 +28,7 @@ tts.to(device)
 
 result_dir = "results"
 
+@spaces.GPU
 def predict(prompt, upload_reference_audio, microphone_reference_audio, reference_audio_type):
     global result_dir
     output_file_path = os.path.join(result_dir, 'output.wav')
@@ -43,7 +44,7 @@ def predict(prompt, upload_reference_audio, microphone_reference_audio, referenc
     )
     return gr.Audio(value=output_file_path, type='filepath')
 
-
+@spaces.GPU
 def main():
     if torch.cuda.is_available():
         device = "cuda"