Minor chnage

app.py

@@ -12,7 +12,7 @@ elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
     device = "mps"
 
 model = GPT(GPTConfig())
-model.load_state_dict(torch.load("gpt2.pt", map_location=torch.device(device)), strict=False)
+model.load_state_dict(torch.load("gpt2.pt", map_location=torch.device(device)))
 
 model.to(device)
 model = torch.compile(model, fullgraph=True, backend="cudagraphs")