Update app.py

app.py

@@ -3,22 +3,25 @@ import gradio as gr
 import outetts
 from outetts.version.v2.interface import _DEFAULT_SPEAKERS
 import torch
-
-model_config = outetts.HFModelConfig_v2(
-    model_path="OuteAI/OuteTTS-0.3-1B",
-    tokenizer_path="OuteAI/OuteTTS-0.3-1B",
-    dtype=torch.bfloat16
-)
-interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)
+import spaces
 
 def get_available_speakers():
     speakers = list(_DEFAULT_SPEAKERS.keys())
     return speakers
 
+@spaces.GPU
 def generate_tts(
         text, temperature, repetition_penalty,
         speaker_selection, reference_audio
     ):
+
+    model_config = outetts.HFModelConfig_v2(
+        model_path="OuteAI/OuteTTS-0.3-1B",
+        tokenizer_path="OuteAI/OuteTTS-0.3-1B",
+        dtype=torch.bfloat16
+    )
+    interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)
+
     """Generate TTS with error handling and new features."""
     try:
         # Validate inputs for custom speaker