from transformers import GPT2Tokenizer, GPT2LMHeadModel

# Load pre-trained model and tokenizer
tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
model = GPT2LMHeadModel.from_pretrained("EleutherAI/gpt-neo-2.7B")

# Input text description for UI
input_text = "Generate a login form with username, password, and a submit button."

# Tokenize input and generate code
input_ids = tokenizer.encode(input_text, return_tensors="pt")
output = model.generate(input_ids, max_length=100)

# Decode and print generated code
generated_code = tokenizer.decode(output[0], skip_special_tokens=True)
print(generated_code)