а здесь происходит вся магия оживления картинки под музыку. для того чтобы попробовать необязательно тренировать что то, можно выбрать из большого списка уже предложенных обученных моделей и музыка тоже доступна прямо в колабе

https://github.com/mikaelalafriz/lucid-sonic-dreams