Skip to content

Commit e069983

Browse files
minor updates
1 parent 1fed1e5 commit e069983

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

riva/client/asr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ def print_streaming(
237237
for i, alternative in enumerate(result.alternatives):
238238
for f in output_file:
239239
f.write(
240-
f"Time {time.time() - start_time:.2f}s: Transcript {i}: {alternative.transcript}\n"
240+
f"Time {time.time() - start_time:.2f}s: Final Transcript {i}: Audio Processed {result.audio_processed}: {alternative.transcript}\n"
241241
)
242242
if word_time_offsets:
243243
for f in output_file:
@@ -269,7 +269,7 @@ def print_streaming(
269269
elif additional_info == 'time':
270270
for f in output_file:
271271
if partial_transcript:
272-
f.write(f">>>Time {time.time():.2f}s: {partial_transcript}\n")
272+
f.write(f">>>Time {time.time() - start_time:.2f}s: {partial_transcript}\n")
273273
else:
274274
for f in output_file:
275275
f.write('----\n')

scripts/asr/riva_streaming_asr_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,13 +61,13 @@ def streaming_transcription_worker(
6161
max_alternatives=args.max_alternatives,
6262
profanity_filter=args.profanity_filter,
6363
enable_automatic_punctuation=args.automatic_punctuation,
64-
verbatim_transcripts=not args.no_verbatim_transcripts,
64+
verbatim_transcripts=not args.verbatim_transcripts,
6565
enable_word_time_offsets=args.word_time_offsets,
6666
model=args.model_name,
6767
),
6868
interim_results=args.interim_results,
6969
)
70-
riva.client.add_word_boosting_to_config(config, args.boosted_lm_words, args.boosted_lm_score)
70+
riva.client.add_word_boosting_to_config(config, args.boosted_words_file, args.boosted_words_score)
7171
for _ in range(args.num_iterations):
7272
with riva.client.AudioChunkFileIterator(
7373
args.input_file,
@@ -80,6 +80,7 @@ def streaming_transcription_worker(
8080
),
8181
input_file=args.input_file,
8282
output_file=output_file,
83+
additional_info='time',
8384
word_time_offsets=args.word_time_offsets,
8485
)
8586
except BaseException as e:

0 commit comments

Comments
 (0)