DHRUV SHEKHAWAT commited on
Commit ·
2eacd00
1
Parent(s): f4fcfa8
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def evaluate(transformer, question, question_mask, max_len, word_map):
|
|
| 52 |
|
| 53 |
|
| 54 |
if load_checkpoint:
|
| 55 |
-
checkpoint = torch.load(ckpt_path)
|
| 56 |
transformer = checkpoint['transformer']
|
| 57 |
|
| 58 |
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
if load_checkpoint:
|
| 55 |
+
checkpoint = torch.load(ckpt_path, map_location=torch.device('cpu'))
|
| 56 |
transformer = checkpoint['transformer']
|
| 57 |
|
| 58 |
|