Update app.py

app.py

@@ -1,8 +1,8 @@
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
+from transformers import GPT2Tokenizer, GPTNeoForCausalLM
 
-# Load pre-trained model and tokenizer
+# Load the tokenizer and model for GPT-Neo
 tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
-model = GPT2LMHeadModel.from_pretrained("EleutherAI/gpt-neo-2.7B")
+model = GPTNeoForCausalLM.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."