Add Zero GPU support

app.py

@@ -2,6 +2,7 @@ import os
 import torchaudio
 import torch
 import gradio as gr
+import spaces
 
 
 from pflow.models.pflow_tts import pflowTTS
@@ -66,7 +67,7 @@ _ = model.eval()
 vocos_44100 = Vocos.from_pretrained(VOCODER44_MODEL_PATH).to(device)
 vocos_22050 = Vocos.from_pretrained(VOCODER22_MODEL_PATH).to(device)
 
-
+@spaces.GPU
 @torch.inference_mode()
 def synthesise(text, ipa, prompt_selection, audio_prompt, temperature, speed):
     print(text, prompt_selection, temperature, speed)