torch.hub.load('pyannote/pyannote-audio', 'dia') не работает в локальном
Я использовал этот код в Google Colab, но он не работает, когда я хочу использовать его в локальном
OWN_FILE = {'audio': 'file.wav'}
pipeline = torch.hub.load('pyannote/pyannote-audio', 'dia')
diarization = pipeline(OWN_FILE)
Вот предупреждения, которые я получаю для torch.hub.load:
FutureWarning: The 's' parameter is deprecated in favor of 'scale', and will be removed in a future release
warnings.warn(msg, FutureWarning)
UserWarning: Model was trained with 4s chunks and is applied on 2s chunks. This might lead to sub-optimal results.
warnings.warn(msg)
У кого-то есть идея использовать пьяннотэ в локальном? заранее спасибо
1 ответ
вы можете просто применить конвейер к аудиофайлу, указав путь к файлу. например:
filePath = "file.wav"
dia_pipeline = torch.hub.load("pyannote/pyannote-audio", "dia")
dia = dia_pipeline({"audio": str(filePath)})