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)})
Другие вопросы по тегам