Нужна помощь с проектом, похожим на pix2code

Всем привет! Делаю дипломный проект, суть которого — на основе нейронной сети переводить изображение web-интерфейса в файлы с кодом HTML и css. Возникли две трудности: 1) нет идей насчет обучающей базы и 2) на каком облачном сервере лучше обучать сетку? Первую проблему можно решить в лоб, взяв обучающую базу из оригинального проекта: https://github.com/tonybeltramelli/pix2code , но мне нужна генерация css-файлов, а в том обучающем материале — только bootstrap. Буду рад вашей помощи.