Update app.py

app.py

@@ -9,7 +9,7 @@ tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
 
 # Load the saved GPT Neo model from the local checkpoint (adjust the path)
 model_path = "elapt1c/ElapticAI-1a"  # Replace with your desired GPT Neo model path
-model = GPTNeoForCausalLM.from_pretrained(model_path)
+model = GPTNeoForCausalLM.from_pretrained(model_path, ignore_mismatched_sizes=True)
 
 # Move model to appropriate device (GPU if available, otherwise CPU)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")