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

sarane22 · web-flow · commit f7d5e3e3dc8e · 2024-11-08T22:47:10.000+05:30
Logging VAD states
diff --git a/common b/common
@@ -1 +1 @@
-Subproject commit 1c7da5aed4e4df3a296d2672379c5099a193aaae
+Subproject commit d73d7c13d3e291aace10f619a8f0e6fc6be78156
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
