Update app.py

app.py

@@ -107,7 +107,7 @@ def tsaigpt(start:str , max_new_tokens = 300, num_samples =2, tokeniser= token):
     device_type = 'cuda' if 'cuda' in device else 'cpu' # for later use in torch.autocast
     ptdtype = {'float32': torch.float32, 'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
     ctx = nullcontext() if device_type == 'cpu' else torch.amp.autocast(device_type=device_type, dtype=ptdtype)
-    checkpoint_path = Path("out/redpajama/iter-023999-ckpt.pth")
+    checkpoint_path = Path("out/redpajama/iter-031997-ckpt.pth")
     config = Config.from_name(model_name)
     model = GPT(config)