GunaKoppula commited on
Commit
5ac1c30
1 Parent(s): 1fe79f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ itos = { i:ch for i,ch in enumerate(chars) }
18
  encode = lambda s: [stoi[c] for c in s]
19
  decode = lambda l: ''.join([itos[i] for i in l])
20
 
21
- model = GPTLanguageModel(vocab_size)
22
  model.load_state_dict(torch.load('saved_model.pth', map_location=cfg.device))
23
  m = model.to(cfg.device)
24
 
 
18
  encode = lambda s: [stoi[c] for c in s]
19
  decode = lambda l: ''.join([itos[i] for i in l])
20
 
21
+ model = GPTModel(vocab_size)
22
  model.load_state_dict(torch.load('saved_model.pth', map_location=cfg.device))
23
  m = model.to(cfg.device)
24