Capstone04 commited on
Commit
57d6a9a
·
verified ·
1 Parent(s): 7240c29

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. asr_diarization/pipeline.py +5 -0
asr_diarization/pipeline.py CHANGED
@@ -2,6 +2,11 @@ import os
2
  import json
3
  import torch
4
  import torchaudio
 
 
 
 
 
5
  import noisereduce as nr
6
  from pyannote.audio import Pipeline
7
  from transformers import WhisperProcessor, WhisperForConditionalGeneration, pipeline as hf_pipeline
 
2
  import json
3
  import torch
4
  import torchaudio
5
+ try:
6
+ from torchaudio.io import AudioMetaData
7
+ torchaudio.AudioMetaData = AudioMetaData
8
+ except ImportError:
9
+ pass
10
  import noisereduce as nr
11
  from pyannote.audio import Pipeline
12
  from transformers import WhisperProcessor, WhisperForConditionalGeneration, pipeline as hf_pipeline