Update app.py

app.py

@@ -29,7 +29,10 @@ enc = tiktoken.Encoding(
 # Load model from checkpoint
 model_save_path = 'tuned_ckpt.pt'
 if os.path.exists(model_save_path):
-    model = torch.load(model_save_path, map_location=device)
+    checkpoint = torch.load(model_save_path, map_location=device)
+    gptconf = GPTConfig(**checkpoint['model_args'])
+    model = GPT(gptconf)
+    model.load_state_dict(checkpoint['model'])
 else:
     raise FileNotFoundError(f"Model file {model_save_path} not found")