Update app.py

app.py

@@ -6,6 +6,7 @@ from tts import StepAudioTTS
 from tokenizer import StepAudioTokenizer
 import os
 import tempfile
+import spaces
 
 class StepAudioDemo:
     def __init__(self):
@@ -23,6 +24,7 @@ class StepAudioDemo:
         self.temp_dir = tempfile.mkdtemp()
         print("Models loaded and ready!")
 
+    @spaces.GPU
     def generate_tts(self, text, speaker_name):
         """Generate TTS audio"""
         try:
@@ -33,6 +35,7 @@ class StepAudioDemo:
         except Exception as e:
             return f"Error generating audio: {str(e)}"
 
+    @spaces.GPU
     def generate_clone(self, text, prompt_audio, prompt_text):
         """Generate cloned voice audio"""
         try: