Automate UI component creation using a ML model

Hey guys
I am building a plugin to automate the creation of components.
For now it’s just the basic components like buttons, labels, input fields and so on. So theres a prompt to enter the requirement and for that i initially created embeddings and the serialised those and fed them to a seq2seq model.
Now i’m thinking of using a gpt2 model and creating prompts based on the dataset (which is on Figma actually and I got through the Figma API) and then use that to create pairs of prompts and their corresponding json.
I’m kinda stuck​:sob: with this so would love to hear your thoughts and opinions on how this could be executed better. If there are any other similar tools minus the whole design tools you find on the internet and instead plugins like this please do let me know :slight_smile: