Update tts.py

tts.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import torchaudio
-import spaces  # Import spaces module for Zero-GPU
+import spaces
 from tortoise.api import TextToSpeech
 from tortoise.utils.audio import load_audio
 
@@ -11,8 +11,9 @@ os.makedirs("outputs", exist_ok=True)
 # Create a global TTS model instance
 tts_model = None
 
-@spaces.GPU  # Add spaces.GPU decorator for Zero-GPU support
-async def generate_speech(text, voice_preset="random", voice_file_path=None):
+# Synchronous function with GPU decorator
+@spaces.GPU
+def _generate_speech_gpu(text, voice_preset="random", voice_file_path=None):
     global tts_model
     
     try:
@@ -49,4 +50,9 @@ async def generate_speech(text, voice_preset="random", voice_file_path=None):
         
     except Exception as e:
         print(f"Error generating speech: {str(e)}")
-        raise
+        raise
+
+# Async wrapper that calls the GPU function
+async def generate_speech(text, voice_preset="random", voice_file_path=None):
+    # Call the GPU-decorated function
+    return _generate_speech_gpu(text, voice_preset, voice_file_path)