Update PrateritumGPT.py

PrateritumGPT.py

@@ -147,7 +147,7 @@ loss_fn = nn.CrossEntropyLoss()
 optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
 
 try:
-    model.load_state_dict(torch.load("data/PrateritumGPT.pth"))
+    model.load_state_dict(torch.load("data_PrateritumGPT.pth"))
     print("Sucessfully loaded model.")
 except:
     pass
@@ -220,4 +220,4 @@ for epoch in range(epochs):
 
     print(f"Epoch {epoch + 1}/{epochs}, Loss: {total_loss / len(train_loader)}")
 
-torch.save(model.state_dict(), "data/PrateritumGPT.pth")
+torch.save(model.state_dict(), "data_PrateritumGPT.pth")