enable zerogpu acceleration

app.py

@@ -10,6 +10,7 @@ import tempfile
 import gradio as gr
 import torch
 from pathlib import Path
+import spaces
 
 # Add current directory to Python path for local zipvoice package
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
@@ -101,6 +102,7 @@ def load_models_and_components(model_name: str):
             model_config["feature"]["sampling_rate"])
 
 
+@spaces.GPU
 def synthesize_speech_gradio(
     text: str,
     prompt_audio_file,
@@ -125,7 +127,7 @@ def synthesize_speech_gradio(
         # Load models and components
         model, tokenizer, vocoder, feature_extractor, sampling_rate = load_models_and_components(model_name)
 
-        device = torch.device("cpu")
+        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
         # Save uploaded audio to temporary file
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_audio: