Update app.py

app.py

@@ -348,7 +348,7 @@ class GPT2Config(object):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 config = GPT2Config()
 model = GPT2LMHeadModel(config)
-state_dict = torch.load(r'C:\vision_model\gpt-2-Pytorch\test\gpt_today\weights\epoch_1.pth', map_location='cpu' if not torch.cuda.is_available() else None)
+state_dict = torch.load(r'epoch_5.pth', map_location='cpu' if not torch.cuda.is_available() else None)
 model = load_weight(model, state_dict)
 model.to(device)
 print(model)