vinay0123 commited on
Commit
30e9439
·
verified ·
1 Parent(s): 506fca5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ class GPTModel(nn.Module):
64
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
65
  model = GPTModel(tokenizer.vocab_size).to(device)
66
 
67
- def load_model(model, path=r"C:\Users\nandu\Documents\travis\models\gpt_model.pth"):
68
  if os.path.exists(path):
69
  model.load_state_dict(torch.load(path, map_location=device))
70
  model.eval()
 
64
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
65
  model = GPTModel(tokenizer.vocab_size).to(device)
66
 
67
+ def load_model(model, path="gpt_model.pth"):
68
  if os.path.exists(path):
69
  model.load_state_dict(torch.load(path, map_location=device))
70
  model.eval()