From 18b1449d1ff06c469e54190508c4f6be05c73d85 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期二, 05 十二月 2023 22:04:14 +0800
Subject: [PATCH] 分角色语音识别支持更多的模型

---
 funasr/utils/whisper_utils/audio.py |    5 ++++-
 1 files changed, 4 insertions(+), 1 deletions(-)

diff --git a/funasr/utils/whisper_utils/audio.py b/funasr/utils/whisper_utils/audio.py
index 004bd0d..6dd4cb1 100644
--- a/funasr/utils/whisper_utils/audio.py
+++ b/funasr/utils/whisper_utils/audio.py
@@ -1,8 +1,11 @@
 import os
 from functools import lru_cache
 from typing import Union
+try:
+    import ffmpeg
+except:
+    print("Please Requires the ffmpeg CLI and `ffmpeg-python` package to be installed.")
 
-import ffmpeg
 import numpy as np
 import torch
 import torch.nn.functional as F

--
Gitblit v1.9.1