Update app.py

app.py

@@ -72,7 +72,12 @@ def download_audio(url):
         # 转换数据类型为int16
         data = np.asarray(data * 32767, dtype=np.int16)
         return sr, data
-    
+
+def is_audio_too_long(audio_tuple):
+    sample_rate, audio_data = audio_tuple
+    audio_length = audio_data.shape[0] / sample_rate
+    return audio_length > 60
+
 def infer(inMic, inAudio, transform, slice_db, lm, sid, dev):
     if inAudio != None:
         sampling_rate, inaudio = inAudio
@@ -81,7 +86,8 @@ def infer(inMic, inAudio, transform, slice_db, lm, sid, dev):
             sampling_rate, inaudio=download_audio(inMic)
         else:
             return "请上传一段音频后再次尝试", None
-
+    if is_audio_too_long((sampling_rate, inaudio)):
+        return "音频超过60s", None
     print("start inference")
     start_time = time.time()
     # 预处理，重编码