spaces conf

kokoro_tts.py

@@ -4,6 +4,21 @@ import numpy as np
 import logging
 
 
+try:
+    import spaces
+
+    @spaces.GPU(duration=60)
+    def get_generator(pipeline, text, voice, speed, split_pattern):
+        return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
+
+    logging.info("Running TTS in spaces")
+except:
+    logging.info("Spaces not available")
+
+    def get_generator(pipeline, text, voice, speed, split_pattern):
+        return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
+
+
 def generate_audio(
     text,
     pipeline=KPipeline(lang_code="a"),
@@ -30,7 +45,7 @@ def generate_audio(
     except:
         logging.warning("CUDA not available, using CPU")
 
-    generator = pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
+    generator = get_generator(pipeline, text, voice, speed, split_pattern="\.")
 
     all_audio = []
     segments = list(generator)  # Get total number of segments