Commit
·
1f93eb0
1
Parent(s):
0a8fb56
fix for silence/noise
Browse files- test-paraformer-onnx.py +6 -1
test-paraformer-onnx.py
CHANGED
|
@@ -84,7 +84,12 @@ def main():
|
|
| 84 |
}
|
| 85 |
output_names = ["logits"]
|
| 86 |
|
| 87 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|