Генератор музыки

import magenta

from magenta.models.music_vae import TrainedModel

from magenta.models.music_vae import configs

from magenta.music import midi_io

# Загрузка предобученной модели MusicVAE

config_name = 'cat-mel_2bar_big'

config = configs.CONFIG_MAP[config_name]

checkpoint_dir = '/path/to/checkpoint'

model = TrainedModel(config, batch_size=1, checkpoint_dir_or_path=checkpoint_dir)

# Генерация музыки

generated_sequence = model.sample(n=1, length=80, temperature=1.0)

# Сохранение результата в MIDI файл

midi_io.sequence_proto_to_midi_file(generated_sequence[0], 'generated_music.mid')