Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 
@@ -32,6 +33,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 ljs_model.to(device)
 vctk_model.to(device)
 
+
+@spaces.GPU
 def ljs_forward(text, speaking_rate=1.0):
     inputs = ljs_tokenizer(text, return_tensors="pt")
 
@@ -44,6 +47,7 @@ def ljs_forward(text, speaking_rate=1.0):
     return gr.make_waveform((22050, waveform))
 
 
+@spaces.GPU
 def vctk_forward(text, speaking_rate=1.0, speaker_id=1):
     inputs = vctk_tokenizer(text, return_tensors="pt")