ITO-Master / utils.py
jhtonyKoo's picture
update yt_download
46ec003
raw
history blame
400 Bytes
from pytubefix import YouTube
import librosa
import numpy as np
def download_youtube_audio(url):
# yt = YouTube(url, use_po_token=True)
yt = YouTube(url)
stream = yt.streams.filter(only_audio=True).first()
filename = stream.download()
audio, sr = librosa.load(filename, sr=44100, mono=False)
if audio.ndim == 1:
audio = np.stack([audio, audio])
return audio.T, sr