Update app.py

app.py

@@ -19,12 +19,19 @@ def transcribe(file_upload, microphone):
     model.load_state_dict(state_dict)
     
     # 載入音訊並強制轉單聲道
-    x, sr = torchaudio.load(file, channels_first=True)  # 確保通道優先格式
+    x, sr = torchaudio.load(file, channels_first=True)  # 載入音訊
+    
+    # 新增：音訊長度檢查（插入在此處）
+    MAX_AUDIO_SECONDS = 600  # 10分鐘限制
+    if x.shape[1] / sr > MAX_AUDIO_SECONDS:
+        raise ValueError(f"音訊長度不可超過 {MAX_AUDIO_SECONDS} 秒，當前音訊長度：{x.shape[1]/sr:.1f} 秒")
+    
+    # 單聲道轉換
     if x.shape[0] > 1:
-        x = torch.mean(x, dim=0, keepdim=True)  # 平均所有通道轉單聲道
+        x = torch.mean(x, dim=0, keepdim=True)
     
     # 執行降噪
-    out = model(x[None])[0]  # 增加batch維度
+    out = model(x[None])[0]
     
     # 後處理
     out = out / max(out.abs().max().item(), 1)