Getting VAD states from the riva-server (#102)

Logging VAD states
nvidia-riva · Nov 8, 2024 · f7d5e3e · f7d5e3e
1 parent e35192a
commit f7d5e3e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/common b/common
diff --git a/riva/client/asr.py b/riva/client/asr.py
@@ -248,6 +248,12 @@ def print_streaming(
                 continue
             partial_transcript = ""
             for result in response.results:
+                if result.pipeline_states and len(result.pipeline_states.vad_probabilities) > 0:
+                    vad_prob_logs = "VAD States: "
+                    for vad_state in result.pipeline_states.vad_probabilities:
+                            vad_prob_logs += str(vad_state) + " "
+                    for i, f in enumerate(output_file):
+                        f.write(vad_prob_logs + "\n")
                 if not result.alternatives:
                     continue
                 transcript = result.alternatives[0].transcript