Fix GPU exec

app.py

@@ -2,6 +2,7 @@ import os
 import io
 import spaces
 import tempfile
+import asyncio
 #import soundfile as sf
 import numpy as np
 from pydub import AudioSegment
@@ -99,15 +100,32 @@ def clean(output_markdown):
 
 
 @spaces.GPU
-async def text_to_speech(output_text, voice, speed, lang):
-    stream = kokoro.create_stream(
-        output_text,
-        voice=voice,
-        speed=float(speed),
-        lang=lang
-    )
-    async for samples, sample_rate in stream:
-        yield numpy_to_mp3(samples, sampling_rate=sample_rate)
+def text_to_speech(output_text, voice, speed, lang):
+    async def stream_audio(output_text, voice, speed, lang):
+        stream = kokoro.create_stream(
+            output_text,
+            voice=voice,
+            speed=float(speed),
+            lang=lang
+        )
+        async for samples, sample_rate in stream:
+            yield numpy_to_mp3(samples, sampling_rate=sample_rate)
+    
+    async def run_stream():
+        async for chunk in stream_audio(output_text, voice, speed, lang):
+            yield chunk
+
+    # Create a new event loop for this thread
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    
+    # Use the new loop to run the async generator
+    iterator = run_stream().__aiter__()
+    while True:
+        try:
+            yield loop.run_until_complete(iterator.__anext__())
+        except StopAsyncIteration:
+            break
 
 
 with gr.Blocks() as demo: