csukuangfj commited on
Commit
1f93eb0
·
1 Parent(s): 0a8fb56

fix for silence/noise

Browse files
Files changed (1) hide show
  1. test-paraformer-onnx.py +6 -1
test-paraformer-onnx.py CHANGED
@@ -84,7 +84,12 @@ def main():
84
  }
85
  output_names = ["logits"]
86
 
87
- outputs = sess.run(output_names, input_feed=inputs)
 
 
 
 
 
88
  log_probs = outputs[0].squeeze(0)
89
  y = log_probs.argmax(axis=-1)
90
 
 
84
  }
85
  output_names = ["logits"]
86
 
87
+ try:
88
+ outputs = sess.run(output_names, input_feed=inputs)
89
+ except ONNXRuntimeError:
90
+ print("Input wav is silence or noise")
91
+ return
92
+
93
  log_probs = outputs[0].squeeze(0)
94
  y = log_probs.argmax(axis=-1)
95