Skip to content

Commit d2213b6

Browse files
fix: add tts config as per latest proto
1 parent 83b7ff8 commit d2213b6

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

riva/client/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,6 @@
3636
from riva.client.proto.riva_asr_pb2 import RecognitionConfig, StreamingRecognitionConfig
3737
from riva.client.proto.riva_audio_pb2 import AudioEncoding
3838
from riva.client.proto.riva_nlp_pb2 import AnalyzeIntentOptions
39-
from riva.client.proto.riva_nmt_pb2 import StreamingTranslateSpeechToSpeechConfig, TranslationConfig
39+
from riva.client.proto.riva_nmt_pb2 import StreamingTranslateSpeechToSpeechConfig, TranslationConfig, SynthesizeSpeechConfig
4040
from riva.client.tts import SpeechSynthesisService
4141
from riva.client.nmt import NeuralMachineTranslationClient

scripts/nmt/s2s_mic.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,13 @@ def main() -> None:
9595
),
9696
translation_config = riva.client.TranslationConfig(
9797
target_language_code=args.target_language_code,
98-
)
98+
),
99+
tts_config = riva.client.SynthesizeSpeechConfig(
100+
encoding=1,
101+
sample_rate_hz=44100,
102+
voice_name="English-US.Female-1",
103+
language_code="en-US",
104+
),
99105
)
100106

101107
#riva.client.add_word_boosting_to_config(config, args.boosted_lm_words, args.boosted_lm_score)

0 commit comments

Comments
 (0)