Spaces:
Sleeping
Sleeping
Commit
·
3e2e994
1
Parent(s):
d092760
initialize pipeline outside function
Browse files- kokoro_tts.py +7 -6
kokoro_tts.py
CHANGED
|
@@ -19,9 +19,15 @@ except:
|
|
| 19 |
return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
|
| 20 |
|
| 21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
def generate_audio(
|
| 23 |
text,
|
| 24 |
-
pipeline=KPipeline(lang_code="a"),
|
| 25 |
voice="af_heart",
|
| 26 |
speed=1,
|
| 27 |
save_segments=False,
|
|
@@ -40,11 +46,6 @@ def generate_audio(
|
|
| 40 |
Returns:
|
| 41 |
numpy.ndarray: Combined audio data at 24kHz sample rate
|
| 42 |
"""
|
| 43 |
-
try:
|
| 44 |
-
pipeline = pipeline.to("cuda")
|
| 45 |
-
except:
|
| 46 |
-
logging.warning("CUDA not available, using CPU")
|
| 47 |
-
|
| 48 |
generator = get_generator(pipeline, text, voice, speed, split_pattern="\.")
|
| 49 |
|
| 50 |
all_audio = []
|
|
|
|
| 19 |
return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
|
| 20 |
|
| 21 |
|
| 22 |
+
pipeline = KPipeline(lang_code="a")
|
| 23 |
+
try:
|
| 24 |
+
pipeline = pipeline.to("cuda")
|
| 25 |
+
except:
|
| 26 |
+
logging.warning("CUDA not available, using CPU")
|
| 27 |
+
|
| 28 |
+
|
| 29 |
def generate_audio(
|
| 30 |
text,
|
|
|
|
| 31 |
voice="af_heart",
|
| 32 |
speed=1,
|
| 33 |
save_segments=False,
|
|
|
|
| 46 |
Returns:
|
| 47 |
numpy.ndarray: Combined audio data at 24kHz sample rate
|
| 48 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
generator = get_generator(pipeline, text, voice, speed, split_pattern="\.")
|
| 50 |
|
| 51 |
all_audio = []
|