Spaces:
Paused
Paused
makaveli
commited on
Commit
•
f2683ae
1
Parent(s):
b03c4ad
add tts_service
Browse files
main.py
CHANGED
@@ -10,6 +10,7 @@ from multiprocessing import Process, Manager, Value, Queue
|
|
10 |
|
11 |
from whisper_live.trt_server import TranscriptionServer
|
12 |
from llm_service import MistralTensorRTLLM
|
|
|
13 |
|
14 |
|
15 |
def parse_arguments():
|
@@ -71,6 +72,7 @@ if __name__ == "__main__":
|
|
71 |
|
72 |
transcription_queue = Queue()
|
73 |
llm_queue = Queue()
|
|
|
74 |
|
75 |
|
76 |
whisper_server = TranscriptionServer()
|
@@ -97,9 +99,16 @@ if __name__ == "__main__":
|
|
97 |
args.phi_tokenizer_path,
|
98 |
transcription_queue,
|
99 |
llm_queue,
|
|
|
100 |
)
|
101 |
)
|
102 |
llm_process.start()
|
103 |
|
|
|
|
|
|
|
|
|
|
|
104 |
llm_process.join()
|
105 |
whisper_process.join()
|
|
|
|
10 |
|
11 |
from whisper_live.trt_server import TranscriptionServer
|
12 |
from llm_service import MistralTensorRTLLM
|
13 |
+
from tts_service import WhisperSpeechTTS
|
14 |
|
15 |
|
16 |
def parse_arguments():
|
|
|
72 |
|
73 |
transcription_queue = Queue()
|
74 |
llm_queue = Queue()
|
75 |
+
audio_queue = Queue()
|
76 |
|
77 |
|
78 |
whisper_server = TranscriptionServer()
|
|
|
99 |
args.phi_tokenizer_path,
|
100 |
transcription_queue,
|
101 |
llm_queue,
|
102 |
+
audio_queue,
|
103 |
)
|
104 |
)
|
105 |
llm_process.start()
|
106 |
|
107 |
+
# audio process
|
108 |
+
# tts_runner = WhisperSpeechTTS()
|
109 |
+
# tts_process = multiprocessing.Process(target=tts_runner.run, args=("0.0.0.0", 8888, audio_queue))
|
110 |
+
# tts_process.start()
|
111 |
+
|
112 |
llm_process.join()
|
113 |
whisper_process.join()
|
114 |
+
# tts_process.join()
|