Spaces:
Sleeping
Sleeping
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 |
|