Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import os
 from kokoro import generate
 from models import build_model
+import spaces
 
 # Initialize model and device
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -23,7 +24,7 @@ voices = {
     'af_sky': torch.load("voices/af_sky.pt", weights_only=True)
 }
 
-
+@spaces.GPU
 def parse_voice_formula(formula):
     """Parse the voice formula string and return the combined voice tensor."""
     if not formula.strip():
@@ -55,6 +56,7 @@ def parse_voice_formula(formula):
             
     return weighted_sum
 
+@spaces.GPU
 def get_new_voice(formula):
     try:
         # Parse the formula and get the combined voice tensor
@@ -67,6 +69,7 @@ def get_new_voice(formula):
     except Exception as e:
         raise gr.Error(f"Failed to create voice: {str(e)}")
 
+@spaces.GPU        
 def text_to_speech(text, formula):
     try:
         if not text.strip():