Spaces:
Runtime error
Runtime error
fix type issue
Browse files- ffmpeg_converter_actor.py +2 -2
- respond_to_prompt_actor.py +2 -1
ffmpeg_converter_actor.py
CHANGED
|
@@ -47,12 +47,12 @@ class FFMpegConverterActor:
|
|
| 47 |
async def push_chunk(self, chunk):
|
| 48 |
try:
|
| 49 |
self.input_pipe.write(chunk)
|
| 50 |
-
|
| 51 |
except BrokenPipeError:
|
| 52 |
# If the pipe is broken, restart the process.
|
| 53 |
await self.start_process()
|
| 54 |
self.input_pipe.write(chunk)
|
| 55 |
-
|
| 56 |
|
| 57 |
# def has_processed_all_data(self):
|
| 58 |
# return self.process.poll() is not None
|
|
|
|
| 47 |
async def push_chunk(self, chunk):
|
| 48 |
try:
|
| 49 |
self.input_pipe.write(chunk)
|
| 50 |
+
await self.input_pipe.drain()
|
| 51 |
except BrokenPipeError:
|
| 52 |
# If the pipe is broken, restart the process.
|
| 53 |
await self.start_process()
|
| 54 |
self.input_pipe.write(chunk)
|
| 55 |
+
await self.input_pipe.drain()
|
| 56 |
|
| 57 |
# def has_processed_all_data(self):
|
| 58 |
# return self.process.poll() is not None
|
respond_to_prompt_actor.py
CHANGED
|
@@ -6,6 +6,7 @@ from text_to_speech_service import TextToSpeechService
|
|
| 6 |
from chat_service import ChatService
|
| 7 |
import asyncio
|
| 8 |
# from ray.actor import ActorHandle
|
|
|
|
| 9 |
|
| 10 |
@ray.remote
|
| 11 |
class PromptToLLMActor:
|
|
@@ -78,7 +79,7 @@ class SpeechToSpeakerActor:
|
|
| 78 |
|
| 79 |
@ray.remote
|
| 80 |
class SpeechToConverterActor:
|
| 81 |
-
def __init__(self, input_queue:Queue, ffmpeg_converter_actor:
|
| 82 |
load_dotenv()
|
| 83 |
self.input_queue = input_queue
|
| 84 |
self.ffmpeg_converter_actor = ffmpeg_converter_actor
|
|
|
|
| 6 |
from chat_service import ChatService
|
| 7 |
import asyncio
|
| 8 |
# from ray.actor import ActorHandle
|
| 9 |
+
from ffmpeg_converter_actor import FFMpegConverterActor
|
| 10 |
|
| 11 |
@ray.remote
|
| 12 |
class PromptToLLMActor:
|
|
|
|
| 79 |
|
| 80 |
@ray.remote
|
| 81 |
class SpeechToConverterActor:
|
| 82 |
+
def __init__(self, input_queue:Queue, ffmpeg_converter_actor:FFMpegConverterActor):
|
| 83 |
load_dotenv()
|
| 84 |
self.input_queue = input_queue
|
| 85 |
self.ffmpeg_converter_actor = ffmpeg_converter_actor
|