makaveli commited on
Commit
f2683ae
1 Parent(s): b03c4ad

add tts_service

Browse files
Files changed (1) hide show
  1. main.py +9 -0
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()