Updated model loading method

app.py

@@ -10,11 +10,8 @@ if torch.cuda.is_available():
 elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
     device = "mps"
 
-ckpt = torch.load("gpt2.pt", map_location=torch.device(device))
-config = GPTConfig(**ckpt['model_args'])
-model = GPT(config)
-state_dict = ckpt['model']
-model.load_state_dict(state_dict)
+model = GPT(GPTConfig())
+model.load_state_dict(torch.load("nanogpt.pth", map_location=torch.device(device)), strict=False)
 
 model.to(device)