Spaces:
Sleeping
Sleeping
GunaKoppula
commited on
Commit
•
5ac1c30
1
Parent(s):
1fe79f0
Update app.py
Browse files
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 =
|
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 |
|