fixed modle loading issue.

app.py

@@ -17,7 +17,7 @@ UNTRAINED_MODEL.eval()
 
 # Load fine-tuned model 
 TRAINED_MODEL = GPT(GPTConfig)
-checkpoint = torch.load("log/model_19072.pt", weights_only=False)
+checkpoint = torch.load("model_19072.pt", weights_only=False, map_location=torch.device('cpu'))
 TRAINED_MODEL.load_state_dict(checkpoint["model"])
 TRAINED_MODEL.to(device)
 TRAINED_MODEL.eval()