wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

services(riva): convention tweaks

Browse Source
This commit is contained in:
vipyne
2024-12-09 21:01:43 -06:00
parent 3f9d39329c
commit 9211a37efc
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/pipecat/services/riva.py
View File

@@ -57,7 +57,7 @@ class FastpitchTTSService(TTSService):
self._voice_id = voice_id self._voice_id = voice_id
self._sample_rate = sample_rate self._sample_rate = sample_rate
self._language_code = params.language self._language_code = params.language
self.quality = params.quality self._quality = params.quality
self.set_model_name("fastpitch-hifigan-tts") self.set_model_name("fastpitch-hifigan-tts")
self.set_voice(voice_id) self.set_voice(voice_id)
@@ -79,7 +79,7 @@ class FastpitchTTSService(TTSService):
self._language_code, self._language_code,
sample_rate_hz=self._sample_rate, sample_rate_hz=self._sample_rate,
audio_prompt_file=None, audio_prompt_file=None,
quality=self.quality, quality=self._quality,
custom_dictionary={}, custom_dictionary={},
) )
return responses return responses
@@ -146,7 +146,7 @@ class ParakeetSTTService(STTService):
] ]
auth = riva.client.Auth(None, True, server, metadata) auth = riva.client.Auth(None, True, server, metadata)
self.asr_service = riva.client.ASRService(auth) self._asr_service = riva.client.ASRService(auth)
config = riva.client.StreamingRecognitionConfig( config = riva.client.StreamingRecognitionConfig(
config=riva.client.RecognitionConfig( config=riva.client.RecognitionConfig(
@@ -175,7 +175,7 @@ class ParakeetSTTService(STTService):
self._stop_threshold_eou, self._stop_threshold_eou,
) )
riva.client.add_custom_configuration_to_config(config, self._custom_configuration) riva.client.add_custom_configuration_to_config(config, self._custom_configuration)
self.config = config self._config = config
self._queue = asyncio.Queue() self._queue = asyncio.Queue()
@@ -203,9 +203,9 @@ class ParakeetSTTService(STTService):
await self._response_task await self._response_task
def _response_handler(self): def _response_handler(self):
responses = self.asr_service.streaming_response_generator( responses = self._asr_service.streaming_response_generator(
audio_chunks=self, audio_chunks=self,
streaming_config=self.config, streaming_config=self._config,
) )
for response in responses: for response in responses:
if not response.results: if not response.results: