From faf21f033f28f0c094f4723251d6fe9dc5a229b4 Mon Sep 17 00:00:00 2001
From: huangyuxin <hyxin2014@126.com>
Date: Thu, 7 Apr 2022 08:48:18 +0000
Subject: [PATCH] add duration limitation for asr

---
 paddlespeech/cli/asr/infer.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py
index 1fb4be43..8eea7ff5 100644
--- a/paddlespeech/cli/asr/infer.py
+++ b/paddlespeech/cli/asr/infer.py
@@ -426,6 +426,11 @@ class ASRExecutor(BaseExecutor):
         try:
             audio, audio_sample_rate = soundfile.read(
                 audio_file, dtype="int16", always_2d=True)
+            audio_duration = audio.shape[0] / audio_sample_rate
+            max_duration = 30.0
+            if audio_duration >= max_duration:
+                logger.error("Please input audio file less then 30 seconds.\n")
+                return
         except Exception as e:
             logger.exception(e)
             logger.error(
-- 
GitLab